Sparse - SpTRSV: adding CudaUVMSpace to tpl_spec_avail

kokkos · May 23, 2024 · 4232c50 · 4232c50
1 parent aad73ca
commit 4232c50
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git 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<double>,
                                                   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<float>,
+                                                  Kokkos::LayoutLeft,
+                                                  Kokkos::CudaUVMSpace)
+KOKKOSSPARSE_SPTRSV_SOLVE_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex<double>,
+                                                  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
@@ -59,6 +59,17 @@ KOKKOSSPARSE_SPTRSV_SYMBOLIC_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex<double>,
                                                      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<float>,
+                                                     Kokkos::LayoutLeft,
+                                                     Kokkos::CudaUVMSpace)
+KOKKOSSPARSE_SPTRSV_SYMBOLIC_TPL_SPEC_AVAIL_CUSPARSE(Kokkos::complex<double>,
+                                                     Kokkos::LayoutLeft,
+                                                     Kokkos::CudaUVMSpace)
+
 #endif  // KOKKOSKERNELS_ENABLE_TPL_CUSPARSE
 
 }  // namespace Impl