Searched refs:Nsolves_team (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 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()
172 …Nsolves_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()
[all …]
|