Merge pull request #89 from janciesko/fixes

Several improvements

Author: janciesko

src/core/Kokkos_RemoteSpaces_LocalDeepCopy.hpp

@@ -75,7 +75,7 @@ void KOKKOS_INLINE_FUNCTION local_deep_copy_contiguous(
 #ifdef KRS_ENABLE_MPISPACE
       dst_data_block_t data_block = dst_data_block_t(
           src.data(), dst.impl_map().handle().loc.win,
-          dst.impl_map().handle().loc.offset, dst.span(), src_rank);
+          dst.impl_map().handle().loc.offset, dst.span(), dst_rank);
 #else
       dst_data_block_t data_block =
           dst_data_block_t(dst.data(), src.data(), dst.span(), dst_rank);
@@ -131,7 +131,7 @@ void KOKKOS_INLINE_FUNCTION local_deep_copy_contiguous(
 #ifdef KRS_ENABLE_MPISPACE
     dst_data_block_t data_block = dst_data_block_t(
         src.data(), dst.impl_map().handle().loc.win,
-        dst.impl_map().handle().loc.offset, dst.span(), src_rank);
+        dst.impl_map().handle().loc.offset, dst.span(), dst_rank);
 #else
     dst_data_block_t data_block =
         dst_data_block_t(dst.data(), src.data(), dst.span(), dst_rank);

src/impl/mpispace/Kokkos_MPISpace.cpp

@@ -92,6 +92,7 @@ void *MPISpace::impl_allocate(
       Kokkos::abort("MPISpace only supports symmetric allocation policy.");
     }
   }
+
   using MemAllocFailure =
       Kokkos::Impl::Experimental::RemoteSpacesMemoryAllocationFailure;
   using MemAllocFailureMode = Kokkos::Impl::Experimental::
@@ -178,6 +179,7 @@ void MPISpace::impl_deallocate(
 }
 
 void MPISpace::fence() const {
+  Kokkos::fence();
   for (int i = 0; i < mpi_windows.size(); i++) {
     if (mpi_windows[i] != MPI_WIN_NULL) {
       MPI_Win_flush_all(mpi_windows[i]);