Prechádzať zdrojové kódy

examples/axpy/axpy.c: cleanup memory and data

Nathalie Furmento 14 rokov pred
rodič
commit
1811824552
1 zmenil súbory, kde vykonal 5 pridanie a 0 odobranie
  1. 5 0
      examples/axpy/axpy.c

+ 5 - 0
examples/axpy/axpy.c

@@ -153,7 +153,9 @@ int main(int argc, char **argv)
 
 	starpu_task_wait_for_all();
 
+	starpu_data_unpartition(handle_x, 0);
 	starpu_data_unpartition(handle_y, 0);
+	starpu_data_unregister(handle_x);
 	starpu_data_unregister(handle_y);
 
 	gettimeofday(&end, NULL);
@@ -164,6 +166,9 @@ int main(int argc, char **argv)
 
 	FPRINTF(stderr, "AFTER y[0] = %2.2f (ALPHA = %2.2f)\n", vec_y[0], alpha);
 
+	starpu_free((void *)vec_x);
+	starpu_free((void *)vec_y);
+
 	/* Stop StarPU */
 	starpu_shutdown();