160   const int          Nsolves      = nBlk;  in PCApply_BJKOKKOSKERNELS()  local
163   const int          nnz            = (int)info.nz_used / Nsolves; // fix for variable grid size  in PCApply_BJKOKKOSKERNELS()
172 …Nsolves % Nsolves_team == 0, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONG, "Nsolves.mod(N…  in PCApply_BJKOKKOSKERNELS()
173 …Nsolves == 0, PetscObjectComm((PetscObject)pc), PETSC_ERR_ARG_WRONG, "info.nz_used.mod(Nsolves) !=…  in PCApply_BJKOKKOSKERNELS()
177 …kos::MemoryTraits<Kokkos::Unmanaged>> inv_diag((PetscScalar *)glb_idiag, Nsolves, Nloc); // in cor…  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()
181     jac->batch_b      = new XYType("batch rhs", Nsolves, Nloc);  in PCApply_BJKOKKOSKERNELS()
182     jac->batch_x      = new XYType("batch sol", Nsolves, Nloc);  in PCApply_BJKOKKOSKERNELS()
183     jac->batch_values = new AMatrixValueView("batch values", Nsolves, nnz);  in PCApply_BJKOKKOSKERNELS()
194 … = %d, league size = %d, maxit = %d\n", (int)jac->n, Nloc, Nsolves, nnz, Nsolves_team, Nsolves / N…  in PCApply_BJKOKKOSKERNELS()
196 …"rowOffsets+map", Kokkos::TeamPolicy<>(Nsolves, team_size, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(con…  in PCApply_BJKOKKOSKERNELS()
216 …"prefix sum", Kokkos::TeamPolicy<>(Nsolves / Nsolves_team, 1, 1), KOKKOS_LAMBDA(const team_member …  in PCApply_BJKOKKOSKERNELS()
224 …"copy matrix", Kokkos::TeamPolicy<>(Nsolves /* /batch_sz */, team_size, PCBJKOKKOS_VEC_SIZE), KOKK…  in PCApply_BJKOKKOSKERNELS()
252   KrylovHandleType handle(Nsolves, Nsolves_team, n_iterations, true);  in PCApply_BJKOKKOSKERNELS()
253   handle.Arnoldi_view = Scalar3DViewType("", Nsolves, n_iterations, Nloc + n_iterations + 3);  in PCApply_BJKOKKOSKERNELS()
259 …"map", Kokkos::TeamPolicy<>(Nsolves /* /batch_sz */, -1, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const…  in PCApply_BJKOKKOSKERNELS()