Searched refs:first_matrix (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 75 const int first_matrix = static_cast<int>(member.league_rank()) * _N_team; in operator ()() local
81 …auto d = Kokkos::subview(_D, Kokkos::make_pair(first_matrix, last_matrix),… in operator ()()
82 …auto x = Kokkos::subview(_X, Kokkos::make_pair(first_matrix, last_matrix),… in operator ()()
83 …auto b = Kokkos::subview(_B, Kokkos::make_pair(first_matrix, last_matrix),… in operator ()()
95 …ScratchPadValuesViewType diag(member.team_scratch(1), last_matrix - first_matrix, _diag.extent(1)); in operator ()()
98 …<MemberType>::invoke(member, Kokkos::subview(_diag, Kokkos::make_pair(first_matrix, last_matrix), … in operator ()()
|