Change init and finalise in test

mpi/tests/checkpoints.c

@@ -114,6 +114,7 @@ int pseudotest_checkpoint_template_register(int argc, char* argv[])
 
 int test_checkpoint_submit(int argc, char* argv[])
 {
+	int ret;
 	starpu_data_handle_t handle0, handle1;
 	starpu_mpi_checkpoint_template_t cp_template;
 	int val0 = 0;
@@ -122,7 +123,11 @@ int test_checkpoint_submit(int argc, char* argv[])
 
 	FPRINTF(stderr, "Go\n");
 
-	STARPU_MPI_INIT();
+	ret = starpu_mpi_init_conf(&argc, &argv, 1, MPI_COMM_WORLD, NULL);
+	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init_conf");
+
+	starpu_mpi_comm_size(MPI_COMM_WORLD, &nb_nodes);
+	starpu_mpi_comm_rank(MPI_COMM_WORLD, &me);
 
 	stage+=me;
 
@@ -186,9 +191,9 @@ int test_checkpoint_submit(int argc, char* argv[])
 
 	usleep(150000);
 	fprintf(stderr, "\n\n");
-	sleep(3);
+	starpu_mpi_wait_for_all(MPI_COMM_WORLD);
 	FPRINTF_MPI(stderr, "Bye!\n");
-	starpu_shutdown();
+	starpu_mpi_shutdown();
 
 	return 0;
 }