Blame src/mpi/coll/alltoallw/alltoallw_allcomm_nb.c

Packit Service c5cf8c
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
Packit Service c5cf8c
/*
Packit Service c5cf8c
 *  (C) 2017 by Argonne National Laboratory.
Packit Service c5cf8c
 *      See COPYRIGHT in top-level directory.
Packit Service c5cf8c
 */
Packit Service c5cf8c
Packit Service c5cf8c
#include "mpiimpl.h"
Packit Service c5cf8c
Packit Service c5cf8c
#undef FUNCNAME
Packit Service c5cf8c
#define FUNCNAME MPIR_Alltoallw_allcomm_nb
Packit Service c5cf8c
#undef FCNAME
Packit Service c5cf8c
#define FCNAME MPL_QUOTE(FUNCNAME)
Packit Service c5cf8c
int MPIR_Alltoallw_allcomm_nb(const void *sendbuf, const int sendcounts[], const int sdispls[],
Packit Service c5cf8c
                              const MPI_Datatype sendtypes[], void *recvbuf, const int recvcounts[],
Packit Service c5cf8c
                              const int rdispls[], const MPI_Datatype recvtypes[],
Packit Service c5cf8c
                              MPIR_Comm * comm_ptr, MPIR_Errflag_t * errflag)
Packit Service c5cf8c
{
Packit Service c5cf8c
    int mpi_errno = MPI_SUCCESS;
Packit Service c5cf8c
    MPI_Request req = MPI_REQUEST_NULL;
Packit Service c5cf8c
    MPIR_Request *req_ptr = NULL;
Packit Service c5cf8c
Packit Service c5cf8c
    /* just call the nonblocking version and wait on it */
Packit Service c5cf8c
    mpi_errno =
Packit Service c5cf8c
        MPIR_Ialltoallw(sendbuf, sendcounts, sdispls, sendtypes, recvbuf, recvcounts, rdispls,
Packit Service c5cf8c
                        recvtypes, comm_ptr, &req_ptr);
Packit Service c5cf8c
    if (mpi_errno)
Packit Service c5cf8c
        MPIR_ERR_POP(mpi_errno);
Packit Service c5cf8c
    if (req_ptr)
Packit Service c5cf8c
        req = req_ptr->handle;
Packit Service c5cf8c
Packit Service c5cf8c
    mpi_errno = MPIR_Wait(&req, MPI_STATUS_IGNORE);
Packit Service c5cf8c
    if (mpi_errno)
Packit Service c5cf8c
        MPIR_ERR_POP(mpi_errno);
Packit Service c5cf8c
Packit Service c5cf8c
  fn_exit:
Packit Service c5cf8c
    return mpi_errno;
Packit Service c5cf8c
  fn_fail:
Packit Service c5cf8c
    goto fn_exit;
Packit Service c5cf8c
}