瀏覽代碼

mpi/tests/insert_task_owner.c: needs 2 nodes

Nathalie Furmento 14 年之前
父節點
當前提交
aa6937dbfe
共有 1 個文件被更改,包括 7 次插入0 次删除
  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);