source: XIOS/trunk/extern/src_ep/ep_wait.cpp @ 1034

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

adding src_ep into extern folder

File size: 4.0 KB
Line 
1/*!
2   \file ep_wait.cpp
3   \since 2 may 2016
4
5   \brief Definitions of MPI wait function: MPI_Wait, MPI_Waitall, MPI_Waitsome, MPI_Waitany
6 */
7
8#include "ep_lib.hpp"
9#include <mpi.h>
10#include "ep_declaration.hpp"
11
12using namespace std;
13
14
15
16namespace ep_lib {     
17
18        int MPI_Wait(MPI_Request *request, MPI_Status *status)
19        {
20
21    if(request->type == 1)
22    {
23      ::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request);
24      ::MPI_Status mpi_status;
25      ::MPI_Wait(&mpi_request, &mpi_status);
26
27
28     
29      status->mpi_status = &mpi_status;
30      status->ep_src = request->ep_src;
31      status->ep_tag = request->ep_tag;
32      status->ep_datatype = request->ep_datatype;
33
34      return 0;
35    }
36
37    if(request->type == 2)
38    {
39      int flag = false;
40      MPI_Message message;
41
42      while(!flag)
43      {
44        Message_Check(request->comm);
45        #pragma omp flush
46        MPI_Improbe(request->ep_src, request->ep_tag, request->comm, &flag, &message, status);
47      }
48
49      int count;
50      MPI_Get_count(status, request->ep_datatype, &count);
51      MPI_Mrecv(request->buf, count, request->ep_datatype, &message, status);
52      status->ep_datatype = request->ep_datatype;
53
54      check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2);
55
56      return 0;
57    }
58
59    if(request->type == 3)
60    {
61      ::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request);
62      ::MPI_Status mpi_status;
63      ::MPI_Wait(&mpi_request, &mpi_status);
64     
65      status->mpi_status = new ::MPI_Status(mpi_status);
66      status->ep_src = request->ep_src;
67      status->ep_tag = request->ep_tag;
68      status->ep_datatype = request->ep_datatype;
69
70      int count;
71      MPI_Get_count(status, request->ep_datatype, &count);
72      check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2);
73    }
74          return MPI_SUCCESS;
75        }
76
77
78
79       
80
81
82        int MPI_Waitall(int count, MPI_Request *array_of_requests, MPI_Status *array_of_statuses)
83        {
84    int dest_rank;
85    MPI_Comm_rank(MPI_COMM_WORLD, &dest_rank);
86   
87    int finished = 0;
88    bool finished_index[count];
89
90          for(int i=0; i<count; i++)
91    {
92      finished_index[i] = false;
93    }
94
95    while(finished < count)
96    {
97      for(int i=0; i<count; i++)
98      {
99        if(finished_index[i] == false) // this request has not been tested.
100        {
101          if(array_of_requests[i].type != 2) // isend or imrecv
102          {     
103            MPI_Wait(&array_of_requests[i], &array_of_statuses[i]);
104            if(array_of_requests[i].type == 3)
105            {
106              int check_count;
107              MPI_Get_count(&array_of_statuses[i], array_of_requests[i].ep_datatype, &check_count);
108              check_sum_recv(array_of_requests[i].buf, count, array_of_requests[i].ep_datatype, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, 2);
109            }
110            finished++;
111            finished_index[i] = true;
112          }
113          else // irecv
114          {
115            int flag = false;
116            MPI_Message message;
117           
118            MPI_Improbe(array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, &flag, &message, &array_of_statuses[i]);
119           
120            if(flag)
121            {
122              //printf("dest_rank = %d, Waiting one message with src = %d, tag = %d, buf = %p\n", dest_rank, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].buf);
123              int recv_count;
124              MPI_Get_count(&array_of_statuses[i], array_of_requests[i].ep_datatype, &recv_count);
125              MPI_Mrecv(array_of_requests[i].buf, recv_count, array_of_requests[i].ep_datatype, &message, &array_of_statuses[i]);
126              check_sum_recv(array_of_requests[i].buf, recv_count, array_of_requests[i].ep_datatype, array_of_requests[i].ep_src, array_of_requests[i].ep_tag, array_of_requests[i].comm, 2);
127
128              finished++;
129              finished_index[i] = true;
130            }
131          }
132        }
133      }   
134    }
135          return MPI_SUCCESS;
136        }
137
138
139
140
141
142}
Note: See TracBrowser for help on using the repository browser.