#197: Avoid external synchronization of host/device copies

DARMA-tasking · Jun 10, 2021 · 337cf83 · 337cf83
1 parent 50b157d
commit 337cf83
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/checkpoint/container/view_serialize.h b/src/checkpoint/container/view_serialize.h
@@ -448,7 +448,10 @@ inline void serialize_impl(SerializerT& s, Kokkos::View<T,Args...>& view) {
   if (init) {
     auto host_view = Kokkos::create_mirror_view(view);
     if (s.isPacking()) {
-      Kokkos::deep_copy(host_view, view);
+      // Create and use an execution space to avoid a global Kokkos::fence()
+      auto exec_space = Kokkos::HostSpace{};
+      Kokkos::deep_copy(exec_space, host_view, view);
+      exec_space.fence();
     }
 
     // Serialize the actual data owned by the Kokkos::View
@@ -474,7 +477,10 @@ inline void serialize_impl(SerializerT& s, Kokkos::View<T,Args...>& view) {
     }
 
     if (s.isUnpacking()) {
-      Kokkos::deep_copy(view, host_view);
+      // Create and use an execution space to avoid a global Kokkos::fence()
+      auto exec_space = Kokkos::HostSpace{};
+      Kokkos::deep_copy(exec_space, view, host_view);
+      exec_space.fence();
     }
   }
 }