Searched refs:Nloc (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/bjacobi/bjkokkos/ |
| H A D | bjkokkoskernels.kokkos.cxx | 162 int Nloc = jac->const_block_size; // same grids in PCApply_BJKOKKOSKERNELS() local
177 …oryTraits<Kokkos::Unmanaged>> inv_diag((PetscScalar *)glb_idiag, Nsolves, Nloc); // in correct ord… in PCApply_BJKOKKOSKERNELS()
179 jac->rowOffsets = new IntView("rowOffsets", Nsolves / Nsolves_team, Nloc + 1); // same grids 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()
185 PetscCall(PetscInfo(pc, "Setup indices Nloc=%d, nnz=%d\n", Nloc, nnz)); in PCApply_BJKOKKOSKERNELS()
194 …nz = %d, Nsolves_team = %d, league size = %d, maxit = %d\n", (int)jac->n, Nloc, Nsolves, nnz, Nsol… in PCApply_BJKOKKOSKERNELS()
219 for (int i = 0; i < Nloc; ++i) rowOffsets(graphID, i + 1) += rowOffsets(graphID, i); in PCApply_BJKOKKOSKERNELS()
233 …raphID, rowOffsets(graphID, rowb - start) + i) = d_isrow[aj[i]] - blkID * Nloc; // local" global -… in PCApply_BJKOKKOSKERNELS()
253 handle.Arnoldi_view = Scalar3DViewType("", Nsolves, n_iterations, Nloc + n_iterations + 3); in PCApply_BJKOKKOSKERNELS()
|