diff --git a/sparse/impl/KokkosSparse_sptrsv_symbolic_impl.hpp b/sparse/impl/KokkosSparse_sptrsv_symbolic_impl.hpp index 36ea2d9df8..6c2782b676 100644 --- a/sparse/impl/KokkosSparse_sptrsv_symbolic_impl.hpp +++ b/sparse/impl/KokkosSparse_sptrsv_symbolic_impl.hpp @@ -274,7 +274,7 @@ void lower_tri_symbolic(ExecSpaceIn& space, TriSolveHandle& thandle, Kokkos::parallel_reduce( "check_count host", Kokkos::RangePolicy( - 0, nodes_per_level.extent(0)), + space, 0, nodes_per_level.extent(0)), KOKKOS_LAMBDA(const long i, long& update) { update += nodes_per_level(i); }, @@ -285,8 +285,7 @@ void lower_tri_symbolic(ExecSpaceIn& space, TriSolveHandle& thandle, check_count = 0; // reset Kokkos::parallel_reduce( "check_count device", - Kokkos::RangePolicy( - 0, dnodes_per_level.extent(0)), + Kokkos::RangePolicy(0, dnodes_per_level.extent(0)), KOKKOS_LAMBDA(const long i, long& update) { update += dnodes_per_level(i); }, @@ -740,8 +739,8 @@ void upper_tri_symbolic(ExecutionSpace& space, TriSolveHandle& thandle, check_count = 0; // reset Kokkos::parallel_reduce( "check_count device", - Kokkos::RangePolicy( - 0, dnodes_per_level.extent(0)), + Kokkos::RangePolicy(space, 0, + dnodes_per_level.extent(0)), KOKKOS_LAMBDA(const long i, long& update) { update += dnodes_per_level(i); },