source: XIOS/dev/branch_openmp/extern/src_ep_dev/ep_wait.cpp @ 1374

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

unify type : MPI_Request

File size: 2.5 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
14extern std::list< ep_lib::MPI_Request* > * EP_PendingRequests;
15#pragma omp threadprivate(EP_PendingRequests)
16
17
18
19namespace ep_lib
20{
21 
22  int MPI_Wait(MPI_Request *request, MPI_Status *status)
23  {
24    if(request->type !=1 && request->type !=2 && request->type !=3) 
25    {
26      printf("Error in request type\n");
27 
28      exit(1);
29    }
30
31    while(request->type == 2) Request_Check();
32
33
34    //::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request);
35    ::MPI_Status mpi_status;
36    ::MPI_Wait(static_cast< ::MPI_Request*>(request->mpi_request), &mpi_status);
37
38   
39           
40 
41    // request->mpi_request = mpi_request;
42
43    status->mpi_status = &mpi_status;
44    status->ep_src = request->ep_src;
45    status->ep_tag = request->ep_tag;
46    status->ep_datatype = request->ep_datatype;
47   
48
49
50    return MPI_SUCCESS;
51
52  }   /*end of mpi_wait*/
53
54
55
56
57
58
59  int MPI_Waitall(int count, MPI_Request *array_of_requests, MPI_Status *array_of_statuses)
60  {
61    std::vector<int> finished(count, 0);
62
63    ::MPI_Request* mpi_request = new ::MPI_Request[count];
64    ::MPI_Status* mpi_status = new ::MPI_Status[count];
65
66    while(std::accumulate(finished.begin(), finished.end(), 0) < count)
67    {
68     
69      for(int i=0; i<count; i++)
70      {
71        if(array_of_requests[i].type !=1 && array_of_requests[i].type !=2 && array_of_requests[i].type !=3) 
72        {
73          printf("Error in request type\n");
74 
75          exit(1);
76        }
77       
78        if(array_of_requests[i].type == 2) Request_Check(); 
79        if(array_of_requests[i].type != 2 && finished.at(i) == 0) 
80        {
81          finished.at(i) = 1;
82          //mpi_request[i] = static_cast< ::MPI_Request >(array_of_requests[i].mpi_request);
83          mpi_request[i] = *static_cast< ::MPI_Request*>(array_of_requests[i].mpi_request);
84        }
85      }   
86    }
87
88    ::MPI_Waitall(count, mpi_request, mpi_status);
89
90    for(int i=0; i<count; i++)
91    {
92      array_of_statuses[i].mpi_status = &mpi_status[i];
93      array_of_statuses[i].ep_src = array_of_requests[i].ep_src;
94      array_of_statuses[i].ep_tag = array_of_requests[i].ep_tag;
95      array_of_statuses[i].ep_datatype = array_of_requests[i].ep_datatype;
96    }
97
98    delete[] mpi_request;
99    delete[] mpi_status;
100   
101   
102    return MPI_SUCCESS;
103  }  /* end of mpi_waitall*/
104
105
106}
107
Note: See TracBrowser for help on using the repository browser.