@@ -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();