Changeset 1520 for XIOS/dev/branch_openmp/extern/src_ep_dev/ep_barrier.cpp

Timestamp:

06/04/18 19:25:08 (6 years ago)

Author:

yushan

Message:

save dev. TO DO : test with xios

File:

• XIOS/dev/branch_openmp/extern/src_ep_dev/ep_barrier.cpp (modified) (4 diffs)

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

@@ -9 +9 @@
   int MPI_Barrier(MPI_Comm comm)
   {
-    if(comm.is_intercomm) return MPI_Barrier_intercomm(comm);
+    if(comm->is_intercomm) return MPI_Barrier_intercomm(comm);
 
-    if(comm.is_ep)
+    if(comm->is_ep) 
     {
-      int ep_rank_loc = comm.ep_comm_ptr->size_rank_info[1].first;
-
-      MPI_Barrier_local(comm);
-
-      if(ep_rank_loc == 0)
-      {
-        ::MPI_Comm mpi_comm = to_mpi_comm(comm.mpi_comm);
-
-        ::MPI_Barrier(mpi_comm);
-      }
-
-      MPI_Barrier_local(comm);
-
-      return 0;
-    }
-    else if(comm.mpi_comm != static_cast< ::MPI_Comm*>(MPI_COMM_NULL.mpi_comm))
-    {
-      ::MPI_Comm mpi_comm = to_mpi_comm(comm.mpi_comm);
-      ::MPI_Barrier(mpi_comm);
-      return 0;
+      return MPI_Barrier_intracomm(comm);
     }
 
-    else return 0;
+    
+    return MPI_Barrier_mpi(comm);
+
   }
 
-
-  int MPI_Barrier_intercomm(MPI_Comm comm)
+  int MPI_Barrier_intracomm(MPI_Comm comm)
   {
-
-    int ep_rank_loc = comm.ep_comm_ptr->size_rank_info[1].first;
+    int ep_rank_loc = comm->ep_comm_ptr->size_rank_info[1].first;
 
     MPI_Barrier_local(comm);
@@ -48 +29 @@
     if(ep_rank_loc == 0)
     {
-      ::MPI_Comm mpi_comm = to_mpi_comm(comm.ep_comm_ptr->intercomm->mpi_inter_comm);
+      ::MPI_Comm mpi_comm = to_mpi_comm(comm->mpi_comm);
+
       ::MPI_Barrier(mpi_comm);
     }
@@ -55 +37 @@
 
     return 0;
+  }
 
+  int MPI_Barrier_intercomm(MPI_Comm comm)
+  {
+    MPI_Barrier_local(comm);
+
+    if(comm->ep_comm_ptr->intercomm->size_rank_info[1].first == 0)
+      ::MPI_Barrier(to_mpi_comm(comm->ep_comm_ptr->intercomm->mpi_inter_comm));
+
+    MPI_Barrier_local(comm);
+  }
+
+  int MPI_Barrier_mpi(MPI_Comm comm)
+  {
+    return ::MPI_Barrier(to_mpi_comm(comm->mpi_comm));
   }
 
@@ -61 +57 @@
   int MPI_Barrier_local(MPI_Comm comm)
   {
-    //Message_Check(comm);
-    comm.ep_barrier->wait();
+    comm->ep_barrier->wait();
   }