Lines Matching refs:Nsolves_team

161   int                Nsolves_team = jac->nsolves_team, fill_idx = 0;  in PCApply_BJKOKKOSKERNELS()  local
170 if (Nsolves_team > batch_sz) Nsolves_team = batch_sz; // silently fix this in PCApply_BJKOKKOSKERNELS()
172Nsolves_team == 0, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONG, "Nsolves.mod(Nsolves_tea… in PCApply_BJKOKKOSKERNELS()
179 jac->rowOffsets = new IntView("rowOffsets", Nsolves / Nsolves_team, Nloc + 1); // same grids in PCApply_BJKOKKOSKERNELS()
180 jac->colIndices = new IntView("colIndices", Nsolves / Nsolves_team, nnz); in PCApply_BJKOKKOSKERNELS()
194 …ize = %d, maxit = %d\n", (int)jac->n, Nloc, Nsolves, nnz, Nsolves_team, Nsolves / Nsolves_team, (i… 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()
216 …"prefix sum", Kokkos::TeamPolicy<>(Nsolves / Nsolves_team, 1, 1), KOKKOS_LAMBDA(const team_member … in PCApply_BJKOKKOSKERNELS()
225 …(), start = d_bid_eqOffset[blkID], end = d_bid_eqOffset[blkID + 1], graphID = blkID / Nsolves_team; in PCApply_BJKOKKOSKERNELS()
233 …if (fill_idx && blkID % Nsolves_team == 0) colIndices(graphID, rowOffsets(graphID, rowb - start) +… in PCApply_BJKOKKOSKERNELS()
252 KrylovHandleType handle(Nsolves, Nsolves_team, n_iterations, true); in PCApply_BJKOKKOSKERNELS()
255 …pe>(batch_values, inv_diag, rowOffsets, colIndices, batch_x, batch_b, Nsolves_team, -1, vector_len… in PCApply_BJKOKKOSKERNELS()