Changeset 1642 for XIOS/dev/branch_openmp/extern/src_ep_dev/ep_reduce_scatter.cpp

Timestamp:

01/23/19 10:31:44 (5 years ago)

Author:

yushan

Message:

dev on ADA. add flag switch _usingEP/_usingMPI

File:

• XIOS/dev/branch_openmp/extern/src_ep_dev/ep_reduce_scatter.cpp (modified) (1 diff)

XIOS/dev/branch_openmp/extern/src_ep_dev/ep_reduce_scatter.cpp

@@ -61 +61 @@
 
     int my_recvcount = recvcounts[ep_rank];
-    MPI_Gather_local(&my_recvcount, 1, MPI_INT, local_recvcounts.data(), 0, comm);
-    MPI_Bcast_local(local_recvcounts.data(), num_ep, MPI_INT, 0, comm);
+    MPI_Gather_local(&my_recvcount, 1, EP_INT, local_recvcounts.data(), 0, comm);
+    MPI_Bcast_local(local_recvcounts.data(), num_ep, EP_INT, 0, comm);
 
     int my_displs = std::accumulate(recvcounts, recvcounts+ep_rank, 0);
-    MPI_Gather_local(&my_displs, 1, MPI_INT, local_displs.data(), 0, comm);
-    MPI_Bcast_local(local_displs.data(), num_ep, MPI_INT, 0, comm);
+    MPI_Gather_local(&my_displs, 1, EP_INT, local_displs.data(), 0, comm);
+    MPI_Bcast_local(local_displs.data(), num_ep, EP_INT, 0, comm);