Skip to content

Commit d7dace1

Browse files
authored
Sparse - SpGEMM: labeling spgemm_symbolic in TPL layer a bit more clearly (kokkos#2193)
This just improves the readability of the output from the tools as it now has a symbolic matching the numeric phase. Previously we only had spgemm in the label which is a bit confusing as it could be the whole spgemm time i.e. both symbolic and numeric, additionally we had symbolic in the MKL path but not in cusparse, rocsparse...
1 parent dfacd72 commit d7dace1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sparse/tpls/KokkosSparse_spgemm_symbolic_tpl_spec_decl.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -387,7 +387,7 @@ void spgemm_symbolic_cusparse(KernelHandle *handle, lno_t m, lno_t n, lno_t k,
387387
bool, c_int_view_t row_mapB, \
388388
c_int_view_t entriesB, bool, \
389389
int_view_t row_mapC, bool computeRowptrs) { \
390-
std::string label = "KokkosSparse::spgemm[TPL_CUSPARSE," + \
390+
std::string label = "KokkosSparse::spgemm_symbolic[TPL_CUSPARSE," + \
391391
Kokkos::ArithTraits<SCALAR>::name() + "]"; \
392392
Kokkos::Profiling::pushRegion(label); \
393393
spgemm_symbolic_cusparse(handle->get_spgemm_handle(), m, n, k, row_mapA, \
@@ -549,7 +549,7 @@ void spgemm_symbolic_rocsparse(
549549
bool, c_int_view_t row_mapB, \
550550
c_int_view_t entriesB, bool, \
551551
int_view_t row_mapC, bool) { \
552-
std::string label = "KokkosSparse::spgemm[TPL_ROCSPARSE," + \
552+
std::string label = "KokkosSparse::spgemm_symbolic[TPL_ROCSPARSE," + \
553553
Kokkos::ArithTraits<SCALAR>::name() + "]"; \
554554
Kokkos::Profiling::pushRegion(label); \
555555
spgemm_symbolic_rocsparse(handle->get_spgemm_handle(), m, n, k, \

0 commit comments

Comments
 (0)