Explorar el Código

tests/main/execute_schedule.c: fix call to starpu_variable_data_register and unregister data + free memory

Nathalie Furmento hace 9 años
padre
commit
af504495c1
Se han modificado 1 ficheros con 6 adiciones y 1 borrados
  1. 6 1
      tests/main/execute_schedule.c

+ 6 - 1
tests/main/execute_schedule.c

@@ -79,7 +79,7 @@ int main(int argc, char **argv)
 	for (n = 0; n < N; n++)
 	{
 		t[n] = malloc((1<<20) * sizeof(*(t[n])));
-		starpu_variable_data_register(&h[n], STARPU_MAIN_RAM, (uintptr_t) &t[n], (1<<20) * sizeof(*(t[n])));
+		starpu_variable_data_register(&h[n], STARPU_MAIN_RAM, (uintptr_t) t[n], (1<<20) * sizeof(*(t[n])));
 	}
 
 	for (k = 0; k < K; k++)
@@ -120,6 +120,11 @@ int main(int argc, char **argv)
 	}
 
 	starpu_task_wait_for_all();
+	for (n = 0; n < N; n++)
+	{
+		starpu_data_unregister(h[n]);
+		free(t[n]);
+	}
 
 	starpu_shutdown();