Преглед изворни кода

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

Nathalie Furmento пре 14 година
родитељ
комит
1811824552
1 измењених фајлова са 5 додато и 0 уклоњено
  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();