Merge pull request #76 from nicoleavans/pr

Add irecv wrapper in KokkosComm namespace

Author: cwpearson

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<SendMode>(space, sv, dest, tag, comm);
 }
 
+template <KokkosView RecvView>
+void irecv(RecvView &rv, int src, int tag, MPI_Comm comm, MPI_Request req) {
+  return Impl::irecv(rv, src, tag, comm, req);
+}
+
 template <CommMode SendMode = CommMode::Default, KokkosExecutionSpace ExecSpace, KokkosView SendView>
 void send(const ExecSpace &space, const SendView &sv, int dest, int tag, MPI_Comm comm) {
   return Impl::send<SendMode>(space, sv, dest, tag, comm);
 }
 
 template <KokkosExecutionSpace ExecSpace, KokkosView RecvView>
-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 <KokkosExecutionSpace ExecSpace, KokkosView SendView, KokkosView RecvView>
+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