[1037] | 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 | |
---|
| 12 | using namespace std; |
---|
| 13 | |
---|
| 14 | |
---|
| 15 | |
---|
| 16 | namespace 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 | |
---|
[1053] | 54 | //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2); |
---|
[1037] | 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 | |
---|
[1067] | 70 | //int count; |
---|
| 71 | //MPI_Get_count(status, request->ep_datatype, &count); |
---|
[1053] | 72 | //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2); |
---|
[1037] | 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 | { |
---|
[1067] | 106 | //int check_count; |
---|
| 107 | //MPI_Get_count(&array_of_statuses[i], array_of_requests[i].ep_datatype, &check_count); |
---|
[1053] | 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); |
---|
[1037] | 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]); |
---|
[1053] | 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); |
---|
[1037] | 127 | |
---|
| 128 | finished++; |
---|
| 129 | finished_index[i] = true; |
---|
| 130 | } |
---|
| 131 | } |
---|
| 132 | } |
---|
| 133 | } |
---|
| 134 | } |
---|
| 135 | return MPI_SUCCESS; |
---|
| 136 | } |
---|
| 137 | |
---|
| 138 | |
---|
| 139 | |
---|
| 140 | |
---|
| 141 | |
---|
| 142 | } |
---|