Changeset 1398 for XIOS/dev/branch_openmp/extern/ep_dev/ep_put.cpp

Timestamp:

01/10/18 16:16:44 (6 years ago)

Author:

yushan

Message:

dev EP-RMA : MPI_Rput MPI_Rget MPI_Raccumulate MPI_Rget_accumulate (to be tested)

File:

• XIOS/dev/branch_openmp/extern/ep_dev/ep_put.cpp (modified) (1 diff)

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

@@ -24 +24 @@
   }
 
+  int MPI_Rput(const void *origin_addr, int origin_count, MPI_Datatype origin_datatype, 
+              int target_rank, MPI_Aint target_disp, int target_count, MPI_Datatype target_datatype, MPI_Win win, MPI_Request *request)
+  {
+    int target_mpi_rank   = win.comm.rank_map->at(target_rank).second;
+    int target_local_rank = win.comm.rank_map->at(target_rank).first;
+    int num_ep = win.comm.ep_comm_ptr->size_rank_info[1].second;
+    
+    ::MPI_Request mpi_request;
+
+    if(num_ep==1)
+    {
+      int return_value = ::MPI_Rput(origin_addr, origin_count, to_mpi_type(origin_datatype), target_mpi_rank, to_mpi_aint(target_disp), target_count, to_mpi_type(target_datatype), 
+                           to_mpi_win(win.server_win[target_local_rank]), &mpi_request);
+
+      request->mpi_request = new ::MPI_Request(mpi_request);
+
+      request->ep_datatype = origin_datatype;
+      request->type = 1;
+      return return_value;
+    }
+
+    else
+    {
+      int return_value = ::MPI_Rput(origin_addr, origin_count, to_mpi_type(origin_datatype), target_mpi_rank, to_mpi_aint(target_disp), target_count, to_mpi_type(target_datatype), 
+                         to_mpi_win(win.client_win), &mpi_request);
+
+      request->mpi_request = new ::MPI_Request(mpi_request);
+
+      request->ep_datatype = origin_datatype;
+      request->type = 1;
+      return return_value;
+    }
+  }
+
 }