diff --git a/src/KokkosComm.hpp b/src/KokkosComm.hpp index 84ab90ea..ae4b5cb2 100644 --- a/src/KokkosComm.hpp +++ b/src/KokkosComm.hpp @@ -19,8 +19,10 @@ #include "KokkosComm_collective.hpp" #include "KokkosComm_version.hpp" #include "KokkosComm_isend.hpp" +#include "KokkosComm_irecv.hpp" #include "KokkosComm_recv.hpp" #include "KokkosComm_send.hpp" +#include "KokkosComm_alltoall.hpp" #include "KokkosComm_concepts.hpp" #include "KokkosComm_comm_mode.hpp" @@ -33,14 +35,25 @@ Req isend(const ExecSpace &space, const SendView &sv, int dest, int tag, MPI_Com return Impl::isend(space, sv, dest, tag, comm); } +template +void irecv(RecvView &rv, int src, int tag, MPI_Comm comm, MPI_Request req) { + return Impl::irecv(rv, src, tag, comm, req); +} + template void send(const ExecSpace &space, const SendView &sv, int dest, int tag, MPI_Comm comm) { return Impl::send(space, sv, dest, tag, comm); } template -void recv(const ExecSpace &space, RecvView &sv, int src, int tag, MPI_Comm comm) { - return Impl::recv(space, sv, src, tag, comm); +void recv(const ExecSpace &space, RecvView &rv, int src, int tag, MPI_Comm comm) { + return Impl::recv(space, rv, src, tag, comm); +} + +template +void alltoall(const ExecSpace &space, const SendView &sv, const size_t sendCount, const RecvView &rv, + const size_t recvCount, MPI_Comm comm) { + return Impl::alltoall(space, sv, sendCount, rv, recvCount, comm); } } // namespace KokkosComm