Changeset 1398 for XIOS/dev/branch_openmp/extern/ep_dev/ep_accumulate.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_accumulate.cpp (modified) (2 diffs)

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

@@ -24 +24 @@
   }
 
+  int MPI_Raccumulate(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_Op op, 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_Raccumulate(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_op(op), 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_Raccumulate(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_op(op), 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;
+    }
+  }
+
+
   int MPI_Get_accumulate(const void *origin_addr, int origin_count, MPI_Datatype origin_datatype, void *result_addr, 
                          int result_count, MPI_Datatype result_datatype, int target_rank, MPI_Aint target_disp,
@@ -39 +71 @@
   }
 
+  int MPI_Rget_accumulate(const void *origin_addr, int origin_count, MPI_Datatype origin_datatype, void *result_addr, 
+                         int result_count, MPI_Datatype result_datatype, int target_rank, MPI_Aint target_disp,
+                         int target_count, MPI_Datatype target_datatype, MPI_Op op, 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_Rget_accumulate(origin_addr, origin_count, to_mpi_type(origin_datatype), result_addr, result_count, to_mpi_type(result_datatype), target_mpi_rank, to_mpi_aint(target_disp), target_count,
+                                      to_mpi_type(target_datatype), to_mpi_op(op), 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_Rget_accumulate(origin_addr, origin_count, to_mpi_type(origin_datatype), result_addr, result_count, to_mpi_type(result_datatype), target_mpi_rank, to_mpi_aint(target_disp), target_count,
+                                      to_mpi_type(target_datatype), to_mpi_op(op), 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;
+    }
+  }
+
 }