diff --git a/sparse/impl/KokkosSparse_spgemm_impl_def.hpp b/sparse/impl/KokkosSparse_spgemm_impl_def.hpp index a420a81c90..54e4e228c8 100644 --- a/sparse/impl/KokkosSparse_spgemm_impl_def.hpp +++ b/sparse/impl/KokkosSparse_spgemm_impl_def.hpp @@ -59,6 +59,7 @@ void KokkosSPGEMM::KokkosSPGEMM_symbolic(c_row_view_t rowmapC_) { + Kokkos::Profiling::pushRegion("KokkosSparse::spgemm_symbolic[NATIVE]"); { if (KOKKOSKERNELS_VERBOSE) { std::cout << "SYMBOLIC PHASE" << std::endl; @@ -162,6 +163,7 @@ void KokkosSPGEMMget_spgemm_handle()->set_c_nnz(result_index); Kokkos::deep_copy(row_mapC, h_rmc); Kokkos::fence(); + Kokkos::Profiling::popRegion(); } template