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