Changeset 1504 for XIOS/dev/branch_openmp/extern/ep_dev/main.cpp

Timestamp:

05/28/18 23:25:35 (6 years ago)

Author:

yushan

Message:

MPI_split can deal with discontinuous ranking within a process

File:

• XIOS/dev/branch_openmp/extern/ep_dev/main.cpp (modified) (2 diffs)

XIOS/dev/branch_openmp/extern/ep_dev/main.cpp

@@ -687 +687 @@
 
     //int color = rank%2;
-    int color, remote_leader;
-    if(rank<size-2) {color = 1; remote_leader = size-2;}
-    else {color = 0; remote_leader = 0;}
-    
-    printf("rank = %d, color = %d, remote_leader = %d\n", rank, color, remote_leader);
-  
-    MPI_Comm sub_comm;
-    MPI_Comm_split(comm, color, rank, &sub_comm);
-
-
-    
-    int sub_rank;
-    MPI_Comm_rank(sub_comm, &sub_rank);
+    int tab_color[16]={2,2,2,3,0,1,1,3,2,1,3,0,0,2,0,0};
+    int tab_key[16]={3,11,10,5,6,8,15,7,2,1,9,13,4,14,12,0};
+    int color = tab_color[rank];
+    int key = tab_key[rank];
+  
+    MPI_Comm split_comm;
+    MPI_Comm_split(comm, color, key, &split_comm);
+
+
+    
+    int split_rank;
+    MPI_Comm_rank(split_comm, &split_rank);
+    
+    printf("rank = %d, color = %d, key = %d, split_rank = %d\n", rank, color, key, split_rank);
 
 
@@ -733 +734 @@
 */
     MPI_Barrier(comm);
-    MPI_Comm_free(&sub_comm);
+    MPI_Comm_free(&split_comm);