source: XIOS/dev/branch_openmp/extern/ep_dev/ep_put.cpp @ 1503

Last change on this file since 1503 was 1503, checked in by yushan, 3 years ago

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

File size: 2.4 KB
Line 
1#include "ep_lib.hpp"
2#include <mpi.h>
3#include "ep_declaration.hpp"
4#include "ep_mpi.hpp"
5
6using namespace std;
7
8namespace ep_lib
9{
10
11        int MPI_Put(const void *origin_addr, int origin_count, MPI_Datatype origin_datatype, 
12                  int target_rank, MPI_Aint target_disp, int target_count, MPI_Datatype target_datatype, MPI_Win win)
13  {
14    int target_mpi_rank   = win->comm->ep_rank_map->at(target_rank).second;
15    int target_local_rank = win->comm->ep_rank_map->at(target_rank).first;
16    int num_ep = win->comm->ep_comm_ptr->size_rank_info[1].second;
17    if(num_ep==1)
18      return  ::MPI_Put(origin_addr, origin_count, to_mpi_type(origin_datatype), target_mpi_rank, to_mpi_aint(target_disp), target_count, to_mpi_type(target_datatype), 
19                       to_mpi_win(win->server_win[target_local_rank]));
20
21    else
22        return  ::MPI_Put(origin_addr, origin_count, to_mpi_type(origin_datatype), target_mpi_rank, to_mpi_aint(target_disp), target_count, to_mpi_type(target_datatype), 
23                       to_mpi_win(win->client_win));
24  }
25
26  int MPI_Rput(const void *origin_addr, int origin_count, MPI_Datatype origin_datatype, 
27              int target_rank, MPI_Aint target_disp, int target_count, MPI_Datatype target_datatype, MPI_Win win, MPI_Request *request)
28  {
29    int target_mpi_rank   = win->comm->ep_rank_map->at(target_rank).second;
30    int target_local_rank = win->comm->ep_rank_map->at(target_rank).first;
31    int num_ep = win->comm->ep_comm_ptr->size_rank_info[1].second;
32   
33    ::MPI_Request mpi_request;
34
35    if(num_ep==1)
36    {
37      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), 
38                           to_mpi_win(win->server_win[target_local_rank]), &mpi_request);
39
40      (*request)->mpi_request = new ::MPI_Request(mpi_request);
41
42      (*request)->ep_datatype = origin_datatype;
43      (*request)->type = 1;
44      return return_value;
45    }
46
47    else
48    {
49      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), 
50                         to_mpi_win(win->client_win), &mpi_request);
51
52      (*request)->mpi_request = new ::MPI_Request(mpi_request);
53
54      (*request)->ep_datatype = origin_datatype;
55      (*request)->type = 1;
56      return return_value;
57    }
58  }
59
60}
Note: See TracBrowser for help on using the repository browser.