Searched refs:graphID (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 78 const int graphID = static_cast<int>(member.league_rank()); in operator ()() local
91 TeamVectorCopy1D::invoke(member, Kokkos::subview(_r, graphID, Kokkos::ALL), r); in operator ()()
92 TeamVectorCopy1D::invoke(member, Kokkos::subview(_c, graphID, Kokkos::ALL), c); in operator ()()
217 const int graphID = team.league_rank(); in PCApply_BJKOKKOSKERNELS()
218 rowOffsets(graphID, 0) = 0; in PCApply_BJKOKKOSKERNELS()
219 for (int i = 0; i < Nloc; ++i) rowOffsets(graphID, i + 1) += rowOffsets(graphID, i); in PCApply_BJKOKKOSKERNELS()
225 …_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + 1], graphID = blkID / Nsolves… in PCApply_BJKOKKOSKERNELS()
233 …if (fill_idx && blkID % Nsolves_team == 0) colIndices(graphID, rowOffsets(graphID, rowb - start) +… in PCApply_BJKOKKOSKERNELS()
234 batch_values(blkID, rowOffsets(graphID, rowb - start) + i) = val; in PCApply_BJKOKKOSKERNELS()
|