source: XIOS/dev/branch_yushan_merged/extern/src_ep_dev/ep_wait.cpp @ 1147

/*!
  \file ep_wait.cpp
  \since 2 may 2016

  \brief Definitions of MPI wait function: MPI_Wait, MPI_Waitall, MPI_Waitsome, MPI_Waitany
  */

#include "ep_lib.hpp"
#include <mpi.h>
#include "ep_declaration.hpp"

using namespace std;



namespace ep_lib 
{       

  int MPI_Wait(MPI_Request *request, MPI_Status *status)
  {

    if(request->type == 1)
    {
      ::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request);
      ::MPI_Status mpi_status;
      ::MPI_Wait(&mpi_request, &mpi_status);

      status->mpi_status = &mpi_status;
      status->ep_src = request->ep_src;
      status->ep_tag = request->ep_tag;
      status->ep_datatype = request->ep_datatype;

      return 0;
    }

    if(request->type == 2)
    {
      int flag = false;
      MPI_Message message;

      while(!flag)
      {
        Message_Check(request->comm);
        #pragma omp flush
        MPI_Improbe(request->ep_src, request->ep_tag, request->comm, &flag, &message, status);
      }

      int count;
      MPI_Get_count(status, request->ep_datatype, &count);
      MPI_Mrecv(request->buf, count, request->ep_datatype, &message, status);
      status->ep_datatype = request->ep_datatype;

      //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2);

      return 0;
    }

    if(request->type == 3)
    {
      ::MPI_Request mpi_request = static_cast< ::MPI_Request >(request->mpi_request);
      ::MPI_Status mpi_status;
      ::MPI_Wait(&mpi_request, &mpi_status);

      status->mpi_status = new ::MPI_Status(mpi_status);
      status->ep_src = request->ep_src;
      status->ep_tag = request->ep_tag;
      status->ep_datatype = request->ep_datatype;

      //check_sum_recv(request->buf, count, request->ep_datatype, request->ep_src, request->ep_tag, request->comm, 2);
    }

    return MPI_SUCCESS;

  }   /*end of mpi_wait*/






  int MPI_Waitall(int count, MPI_Request *array_of_requests, MPI_Status *array_of_statuses)
  {
    int dest_rank;
    MPI_Comm_rank(MPI_COMM_WORLD, &dest_rank);

    int finished = 0;
    int finished_index[count];

    for(int i=0; i<count; i++)
    {
      finished_index[i] = false;
    }

    while(finished < count)
    {
      for(int i=0; i<count; i++)
      {
        if(finished_index[i] == false) // this request has not been tested.
        {
          if(array_of_requests[i].type != 2) // isend or imrecv
          {      
            MPI_Wait(&array_of_requests[i], &array_of_statuses[i]);
            finished++;
            finished_index[i] = true;
          }
          else // irecv
          {
            int flag = false;
            MPI_Message message;

            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]);

            if(flag)
            {
              int recv_count;
              MPI_Get_count(&array_of_statuses[i], array_of_requests[i].ep_datatype, &recv_count);
              MPI_Mrecv(array_of_requests[i].buf, recv_count, array_of_requests[i].ep_datatype, &message, &array_of_statuses[i]);
              //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);

              finished++;
              finished_index[i] = true;
            }
          }
        }
      }    
    }
    return MPI_SUCCESS;
  }  /* end of mpi_waitall*/


}