tests: call starpu_data_unregister when data are no longer used

tests/datawizard/mpi_like.c

@@ -153,7 +153,6 @@ static void *thread_func(void *arg)
 		}
 	}
 
-	starpu_data_unregister(thread_data->handle);
 	return NULL;
 }
 
@@ -199,6 +198,11 @@ int main(int argc, char **argv)
 	}
 	starpu_data_release(last_handle);
 
+	for (t = 0; t < NTHREADS; t++)
+	{
+		starpu_data_unregister(problem_data[t].handle);
+	}
+
 	starpu_shutdown();
 
 	return EXIT_SUCCESS;

tests/datawizard/mpi_like_async.c

@@ -278,8 +278,6 @@ static void *thread_func(void *arg)
 	ret = starpu_task_wait_for_all();
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_wait_for_all");
 
-	starpu_data_unregister(thread_data->handle);
-
 	return NULL;
 }
 
@@ -347,6 +345,11 @@ int main(int argc, char **argv)
 	}
 	starpu_data_release(last_handle);
 
+	for (t = 0; t < NTHREADS; t++)
+	{
+		starpu_data_unregister(problem_data[t].handle);
+	}
+
 	starpu_shutdown();
 
 	return EXIT_SUCCESS;