mpid_probe.c

来自「fortran并行计算包」· C语言 代码 · 共 39 行

/*  (C)Copyright IBM Corp.  2007, 2008  */
/**
 * \file src/misc/mpid_probe.c
 * \brief ???
 */
#include "mpidimpl.h"

int MPID_Probe(int source,
               int tag,
               MPID_Comm * comm,
               int context_offset,
               MPI_Status * status)
{
  MPID_Request * rreq;
  MPID_Progress_state state;
  const int context = comm->recvcontext_id + context_offset;

  if (source == MPI_PROC_NULL)
    {
        MPIR_Status_set_procnull(status);
        return MPI_SUCCESS;
    }
  for(;;)
    {
      MPID_Progress_start(&state);
      rreq = MPIDI_Recvq_FU(source, tag, context);
      if (rreq == NULL) MPID_Progress_wait(&state);
      else
        {
          if (status != MPI_STATUS_IGNORE) *status = rreq->status;

          MPID_Request_release(rreq);
          MPID_Progress_end(&state);
          return MPI_SUCCESS;
        }
    }
  return MPI_SUCCESS;
}