Lines Matching refs:blkID

197       const int blkID = team.league_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + 1];
199         if (blkID % Nsolves_team == 0) {                                                        // first matrix on this member
203             rowOffsets(blkID / Nsolves_team, rowb + 1 - start) = n; // save sizes
210         batch_b(blkID, rowb - start) = glb_bdata[rowa];
225       const int blkID = team.league_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + 1], graphID = blkID / Nsolves_team;
233           if (fill_idx && blkID % Nsolves_team == 0) colIndices(graphID, rowOffsets(graphID, rowb - start) + i) = d_isrow[aj[i]] - blkID * Nloc; // local" global - block start
234           batch_values(blkID, rowOffsets(graphID, rowb - start) + i) = val;
260       const int blkID = team.league_rank(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + 1]; // 0
264         glb_xdata[rowa] = batch_x(blkID, rowb - start);
299       for (int blkID = 0; blkID < nBlk; blkID++) {
300         if (handle.get_iteration_host(blkID) > jac->max_nits) {
301           jac->max_nits = handle.get_iteration_host(blkID);
302           mbid          = blkID;
311   for (int blkID = 0; blkID < nBlk; blkID++) {
313       if (jac->batch_target == blkID) {
315           PetscCall(PetscPrintf(PetscObjectComm((PetscObject)A), "    Linear solve %s in %d iterations, batch %" PetscInt_FMT ", species %" PetscInt_FMT "\n", handle.is_converged_host(blkID) ? "converged" : "diverged", handle.get_iteration_host(blkID), blkID % batch_sz, blkID / batch_sz));
316         else PetscCall(PetscPrintf(PetscObjectComm((PetscObject)A), "    Linear solve %s in %d iterations, block %" PetscInt_FMT "\n", handle.is_converged_host(blkID) ? "converged" : "diverged", handle.get_iteration_host(blkID), blkID));
317       } else if (jac->batch_target == -1 && handle.get_iteration_host(blkID) >= count) {
318         jac->max_nits = count = handle.get_iteration_host(blkID);
319         mbid                  = blkID;
321       if (!handle.is_converged_host(blkID)) PetscCall(PetscPrintf(PETSC_COMM_SELF, "ERROR species %d, batch %d did not converge with %d iterations\n", (int)(blkID / batch_sz), (int)blkID % batch_sz, handle.get_iteration_host(blkID)));