From 4232c50ee3a14a024a6751e65ea42633b9ad1e7b Mon Sep 17 00:00:00 2001 From: Luc Berger-Vergiat Date: Thu, 23 May 2024 09:24:17 -0600 Subject: [PATCH] Sparse - SpTRSV: adding CudaUVMSpace to tpl_spec_avail --- .../tpls/KokkosSparse_sptrsv_solve_tpl_spec_avail.hpp | 11 +++++++++++ .../KokkosSparse_sptrsv_symbolic_tpl_spec_avail.hpp | 11 +++++++++++ 2 files changed, 22 insertions(+) diff --git a/sparse/tpls/KokkosSparse_sptrsv_solve_tpl_spec_avail.hpp b/sparse/tpls/KokkosSparse_sptrsv_solve_tpl_spec_avail.hpp index 88f11e7d19..5ada01e22d 100644 --- a/sparse/tpls/KokkosSparse_sptrsv_solve_tpl_spec_avail.hpp +++ b/sparse/tpls/KokkosSparse_sptrsv_solve_tpl_spec_avail.hpp @@ -65,6 +65,17 @@ KOKKOSSPARSE_SPTRSV_SOLVE_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex, Kokkos::LayoutLeft, Kokkos::CudaSpace) +KOKKOSSPARSE_SPTRSV_SOLVE_TPL_SPEC_AVAIL_CUSPARSE(float, Kokkos::LayoutLeft, + Kokkos::CudaUVMSpace) +KOKKOSSPARSE_SPTRSV_SOLVE_TPL_SPEC_AVAIL_CUSPARSE(double, Kokkos::LayoutLeft, + Kokkos::CudaUVMSpace) +KOKKOSSPARSE_SPTRSV_SOLVE_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex, + Kokkos::LayoutLeft, + Kokkos::CudaUVMSpace) +KOKKOSSPARSE_SPTRSV_SOLVE_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex, + Kokkos::LayoutLeft, + Kokkos::CudaUVMSpace) + #endif // KOKKOSKERNELS_ENABLE_TPL_CUSPARSE } // namespace Impl diff --git a/sparse/tpls/KokkosSparse_sptrsv_symbolic_tpl_spec_avail.hpp b/sparse/tpls/KokkosSparse_sptrsv_symbolic_tpl_spec_avail.hpp index 4c88adf757..bdf69cc3d0 100644 --- a/sparse/tpls/KokkosSparse_sptrsv_symbolic_tpl_spec_avail.hpp +++ b/sparse/tpls/KokkosSparse_sptrsv_symbolic_tpl_spec_avail.hpp @@ -59,6 +59,17 @@ KOKKOSSPARSE_SPTRSV_SYMBOLIC_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex, Kokkos::LayoutLeft, Kokkos::CudaSpace) +KOKKOSSPARSE_SPTRSV_SYMBOLIC_TPL_SPEC_AVAIL_CUSPARSE(float, Kokkos::LayoutLeft, + Kokkos::CudaUVMSpace) +KOKKOSSPARSE_SPTRSV_SYMBOLIC_TPL_SPEC_AVAIL_CUSPARSE(double, Kokkos::LayoutLeft, + Kokkos::CudaUVMSpace) +KOKKOSSPARSE_SPTRSV_SYMBOLIC_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex, + Kokkos::LayoutLeft, + Kokkos::CudaUVMSpace) +KOKKOSSPARSE_SPTRSV_SYMBOLIC_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex, + Kokkos::LayoutLeft, + Kokkos::CudaUVMSpace) + #endif // KOKKOSKERNELS_ENABLE_TPL_CUSPARSE } // namespace Impl