source: XIOS/dev/branch_openmp/extern/src_ep_dev/ep_recv.cpp @ 1295

Last change on this file since 1295 was 1295, checked in by yushan, 4 years ago

EP update all

File size: 3.1 KB
Line 
1/*!
2   \file ep_recv.cpp
3   \since 2 may 2016
4
5   \brief Definitions of MPI receive functions: MPI_Recv, MPI_Mrecv, MPI_Irecv, MPI_Imrecv
6 */
7
8
9#include "ep_lib.hpp"
10#include <mpi.h>
11#include "ep_declaration.hpp"
12#include "ep_mpi.hpp"
13
14using namespace std;
15
16extern std::list< ep_lib::MPI_Request* > * EP_PendingRequests;
17#pragma omp threadprivate(EP_PendingRequests)
18
19namespace ep_lib
20{
21 
22
23  int MPI_Recv(void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Status *status)
24  {
25
26    if(!comm.is_ep)
27    {
28      ::MPI_Status mpi_status;
29      ::MPI_Recv(buf, count, to_mpi_type(datatype), src<0? MPI_ANY_SOURCE : src, tag<0? MPI_ANY_TAG: tag, to_mpi_comm(comm.mpi_comm), &mpi_status);
30
31      status->ep_src = src;
32      status->ep_tag = tag;
33      status->ep_datatype = datatype;
34
35      return 0; 
36    }
37
38    Message_Check(comm);
39
40    MPI_Request request;
41    MPI_Irecv(buf, count, datatype, src, tag, comm, &request);
42    MPI_Wait(&request, status);
43
44    return 0;
45  }
46
47
48
49
50  int MPI_Irecv(void *buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Request *request)
51  {
52
53    Debug("MPI_Irecv with EP");
54    int dest_rank;
55    MPI_Comm_rank(comm, &dest_rank);
56   
57
58    if(!comm.is_ep)
59    {
60      ::MPI_Request mpi_request;
61      ::MPI_Irecv(buf, count, to_mpi_type(datatype), src<0? MPI_ANY_SOURCE : src, tag<0? MPI_ANY_TAG: tag, to_mpi_comm(comm.mpi_comm), &mpi_request);
62
63      request->mpi_request = mpi_request;
64      request->ep_src = src;
65      request->ep_datatype = datatype;
66      request->ep_tag = tag;
67    }
68
69    request->mpi_request = MPI_REQUEST_NULL.mpi_request;
70    request->buf = buf;
71    request->comm = comm;
72    request->type = 2;
73
74    request->ep_src = src;
75    request->ep_tag = tag;
76    request->ep_datatype = datatype;
77
78
79
80    /* With Improbe*/
81    Message_Check(comm);
82
83    if(EP_PendingRequests == 0 ) 
84    {
85      EP_PendingRequests = new std::list< MPI_Request* >;
86    }
87
88
89    EP_PendingRequests->push_back(request);
90
91    Request_Check();
92
93    return 0;
94  }
95
96  int MPI_Imrecv(void *buf, int count, MPI_Datatype datatype, MPI_Message *message, MPI_Request *request)
97  {
98    Debug("MPI_Imrecv");
99
100    request->type = 3;
101
102    ::MPI_Request mpi_request;
103    ::MPI_Message mpi_message = static_cast< ::MPI_Message >(message->mpi_message);
104               
105    ::MPI_Imrecv(buf, count, to_mpi_type(datatype), &mpi_message, &mpi_request);
106
107    request->mpi_request = mpi_request;
108    request->ep_datatype = datatype;
109    request->ep_tag = message->ep_tag;
110    request->ep_src = message->ep_src;
111
112    return 0;
113  }
114
115
116  int MPI_Mrecv(void *buf, int count, MPI_Datatype datatype, MPI_Message *message, MPI_Status *status)
117  {
118    Debug("EP Mrecv called\n");
119
120    ::MPI_Status mpi_status;
121    ::MPI_Message mpi_message = static_cast< ::MPI_Message >(message->mpi_message);
122   
123    ::MPI_Mrecv(buf, count, to_mpi_type(datatype), &mpi_message, &mpi_status);
124
125    status->mpi_status = new ::MPI_Status(mpi_status);
126    status->ep_src = message->ep_src;
127    status->ep_datatype = datatype;
128    status->ep_tag = message->ep_tag;
129
130    //check_sum_recv(buf, count, datatype, message->ep_src, message->ep_tag);
131
132    return 0;
133  }
134
135}
136
137
138
Note: See TracBrowser for help on using the repository browser.