Home
last modified time | relevance | path

Searched refs:nBlocks (Results 1 – 3 of 3) sorted by relevance

/petsc/include/petsc/private/
H A Dpcbjkokkosimpl.h40 PetscInt nBlocks; /* total number of blocks */ member
/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkos.kokkos.cxx539 const PetscInt nwork = jac->nwork, nBlk = jac->nBlocks; in PCApply_BJKOKKOS()
922 jac->nBlocks = 0; in PCSetUp_BJKOKKOS()
931 jac->nBlocks += Nf; in PCSetUp_BJKOKKOS()
933 … "%" PetscInt_FMT ") %" PetscInt_FMT " blocks (%" PetscInt_FMT " total)\n", ii, Nf, jac->nBlocks)); in PCSetUp_BJKOKKOS()
935 … "%" PetscInt_FMT ") %" PetscInt_FMT " blocks (%" PetscInt_FMT " total)\n", ii, Nf, jac->nBlocks)); in PCSetUp_BJKOKKOS()
940 …cInt *, Kokkos::LayoutRight, Kokkos::HostSpace> h_block_offsets("block_offsets", jac->nBlocks + 1); in PCSetUp_BJKOKKOS()
953 …, "Add first of %" PetscInt_FMT " blocks with %" PetscInt_FMT " equations\n", jac->nBlocks, nblk)); in PCSetUp_BJKOKKOS()
955 … Add block with %" PetscInt_FMT " equations of %" PetscInt_FMT "\n", idx + 1, nblk, jac->nBlocks)); in PCSetUp_BJKOKKOS()
962 PetscCall(DMCompositeRestoreAccessArray(pack, jac->vec_diag, jac->nBlocks, NULL, subX)); in PCSetUp_BJKOKKOS()
981 …"Diag", Kokkos::TeamPolicy<>(jac->nBlocks, team_size, PCBJKOKKOS_VEC_SIZE), KOKKOS_LAMBDA(const te… in PCSetUp_BJKOKKOS()
[all …]
H A Dbjkokkoskernels.kokkos.cxx159 const PetscInt maxit = jac->ksp->max_it, nBlk = jac->nBlocks; in PCApply_BJKOKKOSKERNELS()