Cyril Roelandt преди 13 години
родител
ревизия
9ba1ab9eb7
променени са 1 файла, в които са добавени 11 реда и са изтрити 0 реда
  1. 11 0
      examples/cg/cg.c

+ 11 - 0
examples/cg/cg.c

@@ -125,6 +125,16 @@ static void generate_random_problem(void)
 	memset(q, 0, n*sizeof(TYPE));
 }
 
+static void free_data(void)
+{
+	starpu_free(A);
+	starpu_free(b);
+	starpu_free(x);
+	starpu_free(r);
+	starpu_free(d);
+	starpu_free(q);
+}
+
 static void register_data(void)
 {
 	starpu_matrix_data_register(&A_handle, 0, (uintptr_t)A, n, n, n, sizeof(TYPE));
@@ -420,6 +430,7 @@ int main(int argc, char **argv)
 
 	starpu_task_wait_for_all();
 	unregister_data();
+	free_data();
 	starpu_helper_cublas_shutdown();
 	starpu_shutdown();