Browse Source

mpi/tests/insert_task_owner.c: needs 2 nodes

Nathalie Furmento 14 years ago
parent
commit
aa6937dbfe
1 changed files with 7 additions and 0 deletions
  1. 7 0
      mpi/tests/insert_task_owner.c

+ 7 - 0
mpi/tests/insert_task_owner.c

@@ -55,6 +55,13 @@ int main(int argc, char **argv)
 	starpu_init(NULL);
 	starpu_mpi_initialize_extended(1, &rank, &size);
 
+        if (size != 2) {
+		if (rank == 0) fprintf(stderr, "We need exactly 2 processes.\n");
+                starpu_mpi_shutdown();
+                starpu_shutdown();
+                return 0;
+        }
+
         if (rank == 0) {
                 starpu_variable_data_register(&data_handlesx0, 0, (uintptr_t)&x0, sizeof(x0));
                 starpu_data_set_rank(data_handlesx0, rank);