Skip to content

Commit dd098a3

Browse files
committed
Stream support for Gauss-Seidel: Symbolic, Numeric, Apply (Twostage)
- Note: everything except for KokkosGraph and sptrsv runs on streams.
1 parent 043376f commit dd098a3

6 files changed

+201
-146
lines changed

common/src/KokkosKernels_Utils.hpp

+2-3
Original file line numberDiff line numberDiff line change
@@ -892,7 +892,7 @@ void permute_block_vector(typename idx_array_type::value_type num_elements,
892892
// TODO BMK: clean this up by removing 1st argument. It is unused but
893893
// its name gives the impression that only num_elements of the vector are
894894
// zeroed, when really it's always the whole thing.
895-
template <class ExecSpaceIn, typename value_array_type, typename MyExecSpace>
895+
template <class ExecSpaceIn, typename value_array_type>
896896
void zero_vector(ExecSpaceIn &exec_space_in,
897897
typename value_array_type::value_type /* num_elements */,
898898
value_array_type &vector) {
@@ -908,8 +908,7 @@ void zero_vector(typename value_array_type::value_type /* num_elements */,
908908
using ne_tmp_t = typename value_array_type::value_type;
909909
ne_tmp_t ne_tmp = ne_tmp_t(0);
910910
MyExecSpace my_exec_space;
911-
zero_vector<MyExecSpace, value_array_type, MyExecSpace>(my_exec_space, ne_tmp,
912-
vector);
911+
zero_vector(my_exec_space, ne_tmp, vector);
913912
}
914913

915914
template <typename v1, typename v2, typename v3>

sparse/impl/KokkosSparse_gauss_seidel_impl.hpp

+4-6
Original file line numberDiff line numberDiff line change
@@ -1547,9 +1547,8 @@ class PointGaussSeidel {
15471547
Permuted_Yvector);
15481548
}
15491549
if (init_zero_x_vector) {
1550-
KokkosKernels::Impl::zero_vector<
1551-
MyExecSpace, scalar_persistent_work_view2d_t, MyExecSpace>(
1552-
my_exec_space, num_cols * block_size, Permuted_Xvector);
1550+
KokkosKernels::Impl::zero_vector(my_exec_space, num_cols * block_size,
1551+
Permuted_Xvector);
15531552
} else {
15541553
KokkosKernels::Impl::permute_block_vector<
15551554
x_value_array_type, scalar_persistent_work_view2d_t,
@@ -1664,9 +1663,8 @@ class PointGaussSeidel {
16641663
Permuted_Yvector);
16651664
}
16661665
if (init_zero_x_vector) {
1667-
KokkosKernels::Impl::zero_vector<
1668-
MyExecSpace, scalar_persistent_work_view2d_t, MyExecSpace>(
1669-
my_exec_space, num_cols, Permuted_Xvector);
1666+
KokkosKernels::Impl::zero_vector(my_exec_space, num_cols,
1667+
Permuted_Xvector);
16701668
} else {
16711669
KokkosKernels::Impl::permute_vector<
16721670
x_value_array_type, scalar_persistent_work_view2d_t,

0 commit comments

Comments
 (0)