Do not use things from namespace Kokkos::Impl (Kokkos::Impl::DeepCopy)

Can achieve the same with Kokkos::deep_copy
kokkos · Feb 2, 2024 · 0ed5a88 · 0ed5a88
1 parent 9331932
commit 0ed5a88
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/sparse/impl/KokkosSparse_coo2crs_impl.hpp b/sparse/impl/KokkosSparse_coo2crs_impl.hpp
@@ -196,8 +196,13 @@ class Coo2Crs {
         reinterpret_cast<UmapType *>(Kokkos::kokkos_malloc<UmapMemorySpace>(
             "m_umaps", m_nrows * sizeof(UmapType)));
 
-    using shallow_copy_to_device =
-        Kokkos::Impl::DeepCopy<UmapMemorySpace, typename Kokkos::HostSpace>;
+    auto shallow_copy_to_device = [](UmapType *dst, UmapType const *src,
+                                     std::size_t cnt) {
+      std::size_t n = cnt / sizeof(UmapType);
+      Kokkos::deep_copy(
+          Kokkos::View<UmapType *, UmapMemorySpace>(dst, n),
+          Kokkos::View<UmapType const *, Kokkos::HostSpace>(src, n));
+    };
 
     UmapType **umap_ptrs = new UmapType *[m_nrows];
     // TODO: use host-level parallel_for with tag rowmapRp1