Lines Matching refs:blkID

633           const int    blkID = team.league_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + 1];
637           bool         print            = monitor && (blkID == view_bid);
640             static_cast<void>(BJSolve_BICG(team, glb_Aai, glb_Aaj, glb_Aaa, d_isrow, d_isicol, work_buff_global, stride_global, nShareVec, work_buff_shared, stride_shared, rtol, atol, dtol, maxit, &d_metadata[blkID], start, end, glb_idiag, glb_bdata, glb_xdata, print));
643             static_cast<void>(BJSolve_TFQMR(team, glb_Aai, glb_Aaj, glb_Aaa, d_isrow, d_isicol, work_buff_global, stride_global, nShareVec, work_buff_shared, stride_shared, rtol, atol, dtol, maxit, &d_metadata[blkID], start, end, glb_idiag, glb_bdata, glb_xdata, print));
706           for (int blkID = 0; blkID < nBlk; blkID++) {
707             if (h_metadata[blkID].its > max_nnit) {
708               jac->max_nits = max_nnit = h_metadata[blkID].its;
710               mbid = blkID;
726       for (int blkID = 0; blkID < nBlk; blkID++) {
727         PetscCall(PetscLogGpuFlops((PetscLogDouble)h_metadata[blkID].flops));
728         PetscCheck(h_metadata[blkID].reason >= 0 || !jac->ksp->errorifnotconverged, PetscObjectComm((PetscObject)pc), PETSC_ERR_CONV_FAILED, "ERROR reason=%s, its=%" PetscInt_FMT ". species %" PetscInt_FMT ", batch %" PetscInt_FMT,
729                    KSPConvergedReasons[h_metadata[blkID].reason], h_metadata[blkID].its, blkID / batch_sz, blkID % batch_sz);
741           for (int blkID = 0; blkID < nBlk; blkID++) {
742             if (h_metadata[blkID].its > jac->max_nits) jac->max_nits = h_metadata[blkID].its;
982           const PetscInt blkID = team.league_rank();
983           Kokkos::parallel_for(Kokkos::TeamThreadRange(team, d_bid_eqOffset[blkID], d_bid_eqOffset[blkID + 1]), [=](const int rowb) {