Fix missing MpiSpace

Signed-off-by: Cedric Chevalier <[email protected]>

Author: cedricchevalier19

src/KokkosComm/mpi/allgather.hpp

@@ -17,7 +17,7 @@
 namespace KokkosComm::mpi {
 
 template <KokkosExecutionSpace ExecSpace, KokkosView SView, KokkosView RView>
-auto iallgather(const ExecSpace &space, const SView sv, RView rv, MPI_Comm comm) -> Req<Mpi> {
+auto iallgather(const ExecSpace &space, const SView sv, RView rv, MPI_Comm comm) -> Req<MpiSpace> {
   using ST = typename SView::non_const_value_type;
   using RT = typename RView::non_const_value_type;
   static_assert(std::is_same_v<ST, RT>, "KokkosComm::mpi::iallgather: View value types must be identical");
@@ -29,7 +29,7 @@ auto iallgather(const ExecSpace &space, const SView sv, RView rv, MPI_Comm comm)
   // Sync: Work in space may have been used to produce view data.
   space.fence("fence before non-blocking all-gather");
 
-  Req<Mpi> req;
+  Req<MpiSpace> req;
   // All ranks send/recv same count
   MPI_Iallgather(data_handle(sv), span(sv), Impl::mpi_type_v<ST>, data_handle(rv), span(sv), Impl::mpi_type_v<RT>, comm,
                  &req.mpi_request());

src/KokkosComm/mpi/alltoall.hpp

@@ -18,7 +18,7 @@
 namespace KokkosComm::mpi {
 
 template <KokkosExecutionSpace ExecSpace, KokkosView SView, KokkosView RView>
-auto ialltoall(const ExecSpace &space, const SView sv, RView rv, int count, MPI_Comm comm) -> Req<Mpi> {
+auto ialltoall(const ExecSpace &space, const SView sv, RView rv, int count, MPI_Comm comm) -> Req<MpiSpace> {
   using ST = typename SView::non_const_value_type;
   using RT = typename RView::non_const_value_type;
   static_assert(std::is_same_v<ST, RT>, "KokkosComm::mpi::ialltoall: View value types must be identical");
@@ -30,7 +30,7 @@ auto ialltoall(const ExecSpace &space, const SView sv, RView rv, int count, MPI_
   // Sync: Work in space may have been used to produce view data.
   space.fence("fence before non-blocking all-gather");
 
-  Req<Mpi> req;
+  Req<MpiSpace> req;
   // All ranks send/recv same count
   MPI_Ialltoall(data_handle(sv), count, Impl::mpi_type_v<ST>, data_handle(rv), count, Impl::mpi_type_v<RT>, comm,
                 &req.mpi_request());

src/KokkosComm/mpi/broadcast.hpp

@@ -17,15 +17,15 @@
 namespace KokkosComm::mpi {
 
 template <KokkosExecutionSpace ExecSpace, KokkosView View>
-auto ibroadcast(const ExecSpace& space, View& v, int root, MPI_Comm comm) -> Req<Mpi> {
+auto ibroadcast(const ExecSpace& space, View& v, int root, MPI_Comm comm) -> Req<MpiSpace> {
   using T = typename View::non_const_value_type;
   Kokkos::Tools::pushRegion("KokkosComm::mpi::ibroadcast");
   fail_if(!is_contiguous(v), "KokkosComm::mpi::ibroadcast: unimplemented for non-contiguous views");
 
   // Sync: Work in space may have been used to produce view data.
   space.fence("fence before non-blocking broadcast");
 
-  Req<Mpi> req;
+  Req<MpiSpace> req;
   MPI_Ibcast(data_handle(v), span(v), Impl::mpi_type_v<T>, root, comm, &req.mpi_request());
   req.extend_view_lifetime(v);