Stream support for Gauss-Seidel: Symbolic, Numeric, Apply (Twostage)

- Note: everything except for KokkosGraph and sptrsv runs on streams.
kokkos · Sep 26, 2023 · dd098a3 · dd098a3
1 parent 043376f
commit dd098a3
Show file tree

Hide file tree

Showing 6 changed files with 201 additions and 146 deletions.
diff --git a/common/src/KokkosKernels_Utils.hpp b/common/src/KokkosKernels_Utils.hpp
@@ -892,7 +892,7 @@ void permute_block_vector(typename idx_array_type::value_type num_elements,
 // TODO BMK: clean this up by removing 1st argument. It is unused but
 // its name gives the impression that only num_elements of the vector are
 // zeroed, when really it's always the whole thing.
-template <class ExecSpaceIn, typename value_array_type, typename MyExecSpace>
+template <class ExecSpaceIn, typename value_array_type>
 void zero_vector(ExecSpaceIn &exec_space_in,
                  typename value_array_type::value_type /* num_elements */,
                  value_array_type &vector) {
@@ -908,8 +908,7 @@ void zero_vector(typename value_array_type::value_type /* num_elements */,
   using ne_tmp_t  = typename value_array_type::value_type;
   ne_tmp_t ne_tmp = ne_tmp_t(0);
   MyExecSpace my_exec_space;
-  zero_vector<MyExecSpace, value_array_type, MyExecSpace>(my_exec_space, ne_tmp,
-                                                          vector);
+  zero_vector(my_exec_space, ne_tmp, vector);
 }
 
 template <typename v1, typename v2, typename v3>

diff --git a/sparse/impl/KokkosSparse_gauss_seidel_impl.hpp b/sparse/impl/KokkosSparse_gauss_seidel_impl.hpp
@@ -1547,9 +1547,8 @@ class PointGaussSeidel {
           Permuted_Yvector);
     }
     if (init_zero_x_vector) {
-      KokkosKernels::Impl::zero_vector<
-          MyExecSpace, scalar_persistent_work_view2d_t, MyExecSpace>(
-          my_exec_space, num_cols * block_size, Permuted_Xvector);
+      KokkosKernels::Impl::zero_vector(my_exec_space, num_cols * block_size,
+                                       Permuted_Xvector);
     } else {
       KokkosKernels::Impl::permute_block_vector<
           x_value_array_type, scalar_persistent_work_view2d_t,
@@ -1664,9 +1663,8 @@ class PointGaussSeidel {
           Permuted_Yvector);
     }
     if (init_zero_x_vector) {
-      KokkosKernels::Impl::zero_vector<
-          MyExecSpace, scalar_persistent_work_view2d_t, MyExecSpace>(
-          my_exec_space, num_cols, Permuted_Xvector);
+      KokkosKernels::Impl::zero_vector(my_exec_space, num_cols,
+                                       Permuted_Xvector);
     } else {
       KokkosKernels::Impl::permute_vector<
           x_value_array_type, scalar_persistent_work_view2d_t,