Changeset 1503 for XIOS/dev/branch_openmp/extern/ep_dev/ep_allgatherv.cpp

Timestamp:

05/28/18 17:28:07 (6 years ago)

Author:

yushan

Message:

rank_map is passed from vector to map, in order to have more flexibility in comm_split

File:

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

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

@@ -92 +92 @@
       {
         int extra = 0;
-        for(int j=0, k=0; j<ep_size, k<comm->rank_map->at(i).first; j++)
-          if(comm->rank_map->at(i).second == comm->rank_map->at(j).second)
+        for(int j=0, k=0; j<ep_size, k<comm->ep_rank_map->at(i).first; j++)
+          if(comm->ep_rank_map->at(i).second == comm->ep_rank_map->at(j).second)
           {
             extra += recvcounts[j];
@@ -99 +99 @@
           }  
 
-        offset = mpi_displs[comm->rank_map->at(i).second] +  extra;
+        offset = mpi_displs[comm->ep_rank_map->at(i).second] +  extra;
 
         memcpy(recvbuf+displs[i]*datasize, tmp_recvbuf+offset*datasize, recvcounts[i]*datasize);