Skip to content

Commit 5ec9b19

Browse files
committed
Sparse: fix CrsMatrix traversal on GPU
Obvious typos and mistakes left-over from CPU development.
1 parent 3485ff0 commit 5ec9b19

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

sparse/impl/KokkosSparse_CrsMatrix_traversal_impl.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ void crsmatrix_traversal_on_gpu(const execution_space& space,
149149
// Wrap user functor with crsmatrix_traversal_functor
150150
int64_t rows_per_thread = 0;
151151
int team_size = 0, vector_length = 0;
152-
const int64_t rows_per_team = crsmatrix_traversal_launch_parameters(A.numRows(), A.nnz(), rows_per_thread, team_size, vector_length);
152+
const int64_t rows_per_team = crsmatrix_traversal_launch_parameters<execution_space>(A.numRows(), A.nnz(), rows_per_thread, team_size, vector_length);
153153
const int nteams = (static_cast<int>(A.numRows()) + rows_per_team - 1) / rows_per_team;
154154
crsmatrix_traversal_functor<execution_space, crsmatrix_type, functor_type> traversal_func(A, func, rows_per_team);
155155

sparse/src/KokkosSparse_CrsMatrix_traversal.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ void crsmatrix_traversal(const execution_space& space, const crsmatrix_type& mat
3737

3838
// Choose between device and host implementation
3939
if constexpr (KokkosKernels::Impl::kk_is_gpu_exec_space<execution_space>()) {
40-
40+
KokkosSparse::Impl::crsmatrix_traversal_on_gpu(space, matrix, functor);
4141
} else {
4242
KokkosSparse::Impl::crsmatrix_traversal_on_host(space, matrix, functor);
4343
}

0 commit comments

Comments
 (0)