int MPID_Iprobe(int source, int tag, MPID_Comm *comm, int context_offset, int *flag, MPI_Status *status) { const int context = comm->recvcontext_id + context_offset; int found = 0; int mpi_errno = MPI_SUCCESS; MPIDI_STATE_DECL(MPID_STATE_MPID_IPROBE); MPIDI_FUNC_ENTER(MPID_STATE_MPID_IPROBE); if (source == MPI_PROC_NULL) { MPIR_Status_set_procnull(status); /* We set the flag to true because an MPI_Recv with this rank will return immediately */ *flag = TRUE; goto fn_exit; } /* Check to make sure the communicator hasn't already been revoked */ if (comm->revoked && MPIR_AGREE_TAG != MPIR_TAG_MASK_ERROR_BITS(tag & ~MPIR_Process.tagged_coll_mask) && MPIR_SHRINK_TAG != MPIR_TAG_MASK_ERROR_BITS(tag & ~MPIR_Process.tagged_coll_mask)) { MPIU_ERR_SETANDJUMP(mpi_errno,MPIX_ERR_REVOKED,"**revoked"); } #ifdef ENABLE_COMM_OVERRIDES if (MPIDI_Anysource_iprobe_fn) { if (source == MPI_ANY_SOURCE) { /* if it's anysource, check shm, then check the network. If still not found, call progress, and check again. */ /* check shm*/ MPIU_THREAD_CS_ENTER(MSGQUEUE,); found = MPIDI_CH3U_Recvq_FU(source, tag, context, status); MPIU_THREAD_CS_EXIT(MSGQUEUE,); if (!found) { /* not found, check network */ mpi_errno = MPIDI_Anysource_iprobe_fn(tag, comm, context_offset, &found, status); if (mpi_errno) MPIU_ERR_POP(mpi_errno); if (!found) { /* still not found, make some progress*/ mpi_errno = MPIDI_CH3_Progress_poke(); if (mpi_errno) MPIU_ERR_POP(mpi_errno); /* check shm again */ MPIU_THREAD_CS_ENTER(MSGQUEUE,); found = MPIDI_CH3U_Recvq_FU(source, tag, context, status); MPIU_THREAD_CS_EXIT(MSGQUEUE,); if (!found) { /* check network again */ mpi_errno = MPIDI_Anysource_iprobe_fn(tag, comm, context_offset, &found, status); if (mpi_errno) MPIU_ERR_POP(mpi_errno); } } }
int MPID_Probe(int source, int tag, MPID_Comm * comm, int context_offset, MPI_Status * status) { MPID_Progress_state progress_state; const int context = comm->recvcontext_id + context_offset; int mpi_errno = MPI_SUCCESS; MPIDI_STATE_DECL(MPID_STATE_MPID_PROBE); MPIDI_FUNC_ENTER(MPID_STATE_MPID_PROBE); if (source == MPI_PROC_NULL) { MPIR_Status_set_procnull(status); goto fn_exit; } #ifdef ENABLE_COMM_OVERRIDES if (MPIDI_Anysource_iprobe_fn) { if (source == MPI_ANY_SOURCE) { /* if it's anysource, loop while checking the shm recv queue and iprobing the netmod, then do a progress test to make some progress. */ do { int found; MPIU_THREAD_CS_ENTER(MSGQUEUE,); found = MPIDI_CH3U_Recvq_FU(source, tag, context, status); MPIU_THREAD_CS_EXIT(MSGQUEUE,); if (found) goto fn_exit; mpi_errno = MPIDI_Anysource_iprobe_fn(tag, comm, context_offset, &found, status); if (mpi_errno) MPIU_ERR_POP(mpi_errno); if (found) goto fn_exit; MPIU_THREAD_CS_YIELD(ALLFUNC,); mpi_errno = MPIDI_CH3_Progress_test(); if (mpi_errno) MPIU_ERR_POP(mpi_errno); } while (1); } else { /* it's not anysource, see if this is for the netmod */ MPIDI_VC_t * vc; MPIDI_Comm_get_vc_set_active(comm, source, &vc); if (vc->comm_ops && vc->comm_ops->iprobe) { /* netmod has overridden iprobe */ do { int found; mpi_errno = vc->comm_ops->iprobe(vc, source, tag, comm, context_offset, &found, status); if (mpi_errno) MPIU_ERR_POP(mpi_errno); if (found) goto fn_exit; MPIU_THREAD_CS_YIELD(ALLFUNC,); mpi_errno = MPIDI_CH3_Progress_test(); if (mpi_errno) MPIU_ERR_POP(mpi_errno); } while (1); } /* fall-through to shm case */ } }