Home
last modified time | relevance | path

Searched refs:blkID (Results 1 – 2 of 2) sorted by relevance

/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkoskernels.kokkos.cxx197 …const int blkID = team.league_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + … in PCApply_BJKOKKOSKERNELS()
199 …if (blkID % Nsolves_team == 0) { // first m… in PCApply_BJKOKKOSKERNELS()
203 rowOffsets(blkID / Nsolves_team, rowb + 1 - start) = n; // save sizes in PCApply_BJKOKKOSKERNELS()
210 batch_b(blkID, rowb - start) = glb_bdata[rowa]; in PCApply_BJKOKKOSKERNELS()
225 …const int blkID = team.league_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + … 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()
260 …const int blkID = team.league_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + … in PCApply_BJKOKKOSKERNELS()
264 glb_xdata[rowa] = batch_x(blkID, rowb - start); in PCApply_BJKOKKOSKERNELS()
299 for (int blkID = 0; blkID < nBlk; blkID++) { in PCApply_BJKOKKOSKERNELS() local
[all …]
H A Dbjkokkos.kokkos.cxx633 …const int blkID = team.league_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID… in PCApply_BJKOKKOS()
637 bool print = monitor && (blkID == view_bid); in PCApply_BJKOKKOS()
640 …ec, work_buff_shared, stride_shared, rtol, atol, dtol, maxit, &d_metadata[blkID], start, end, glb_… in PCApply_BJKOKKOS()
643 …ec, work_buff_shared, stride_shared, rtol, atol, dtol, maxit, &d_metadata[blkID], start, end, glb_… in PCApply_BJKOKKOS()
706 for (int blkID = 0; blkID < nBlk; blkID++) { in PCApply_BJKOKKOS() local
707 if (h_metadata[blkID].its > max_nnit) { in PCApply_BJKOKKOS()
708 jac->max_nits = max_nnit = h_metadata[blkID].its; in PCApply_BJKOKKOS()
710 mbid = blkID; in PCApply_BJKOKKOS()
726 for (int blkID = 0; blkID < nBlk; blkID++) { in PCApply_BJKOKKOS() local
727 PetscCall(PetscLogGpuFlops((PetscLogDouble)h_metadata[blkID].flops)); in PCApply_BJKOKKOS()
[all …]