diff --git a/batched/KokkosBatched_Util.hpp b/batched/KokkosBatched_Util.hpp index 9078281e59..71c40482d6 100644 --- a/batched/KokkosBatched_Util.hpp +++ b/batched/KokkosBatched_Util.hpp @@ -629,8 +629,7 @@ KOKKOS_INLINE_FUNCTION auto subview_wrapper(ViewType v, IdxType1 i1, #if KOKKOS_VERSION < 40099 template KOKKOS_INLINE_FUNCTION auto subview_wrapper(ViewType v, IdxType1 i1, - Kokkos::Impl::ALL_t i2, - Kokkos::Impl::ALL_t i3, + Kokkos::ALL_t i2, Kokkos::ALL_t i3, const BatchLayout::Left &layout_tag, const Trans::Transpose) { auto sv_nt = subview_wrapper(v, i1, i3, i2, layout_tag); @@ -674,7 +673,7 @@ KOKKOS_INLINE_FUNCTION auto subview_wrapper( #if KOKKOS_VERSION < 40099 template KOKKOS_INLINE_FUNCTION auto subview_wrapper( - ViewType v, IdxType1 i1, Kokkos::Impl::ALL_t i2, Kokkos::Impl::ALL_t i3, + ViewType v, IdxType1 i1, Kokkos::ALL_t i2, Kokkos::ALL_t i3, const BatchLayout::Right &layout_tag, const Trans::Transpose &) { auto sv_nt = subview_wrapper(v, i1, i3, i2, layout_tag); diff --git a/sparse/impl/KokkosSparse_coo2crs_impl.hpp b/sparse/impl/KokkosSparse_coo2crs_impl.hpp index d00a6f34a9..f11032903d 100644 --- a/sparse/impl/KokkosSparse_coo2crs_impl.hpp +++ b/sparse/impl/KokkosSparse_coo2crs_impl.hpp @@ -196,8 +196,13 @@ class Coo2Crs { reinterpret_cast(Kokkos::kokkos_malloc( "m_umaps", m_nrows * sizeof(UmapType))); - using shallow_copy_to_device = - Kokkos::Impl::DeepCopy; + auto shallow_copy_to_device = [](UmapType *dst, UmapType const *src, + std::size_t cnt) { + std::size_t nn = cnt / sizeof(UmapType); + Kokkos::deep_copy( + Kokkos::View(dst, nn), + Kokkos::View(src, nn)); + }; UmapType **umap_ptrs = new UmapType *[m_nrows]; // TODO: use host-level parallel_for with tag rowmapRp1