Home
last modified time | relevance | path

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

/petsc/src/ksp/pc/impls/bjacobi/bjkokkos/
H A Dbjkokkos.kokkos.cxx46 KOKKOS_INLINE_FUNCTION PetscErrorCode MatMult(const team_member team, const PetscInt *glb_Aai, cons… in MatMult() argument
50 int n = glb_Aai[rowa + 1] - glb_Aai[rowa]; in MatMult()
51 const PetscInt *aj = glb_Aaj + glb_Aai[rowa]; // global in MatMult()
52 const PetscScalar *aa = glb_Aaa + glb_Aai[rowa]; in MatMult()
62 … PetscErrorCode MatMultTranspose(const team_member team, const PetscInt *glb_Aai, const PetscInt *… in MatMultTranspose() argument
68 int n = glb_Aai[rowa + 1] - glb_Aai[rowa]; in MatMultTranspose()
69 const PetscInt *aj = glb_Aaj + glb_Aai[rowa]; // global in MatMultTranspose()
70 const PetscScalar *aa = glb_Aaa + glb_Aai[rowa]; in MatMultTranspose()
88 …ION PetscErrorCode BJSolve_TFQMR(const team_member team, const PetscInt *glb_Aai, const PetscInt *… in BJSolve_TFQMR() argument
201 static_cast<void>(MatMult(team, glb_Aai, glb_Aaj, glb_Aaa, r, ic, start, end, T, V)); in BJSolve_TFQMR()
[all …]
H A Dbjkokkoskernels.kokkos.cxx155 …c, const PetscScalar *glb_bdata, PetscScalar *glb_xdata, const PetscInt *glb_Aai, const PetscInt *… in PCApply_BJKOKKOSKERNELS() argument
202 int n = glb_Aai[rowa + 1] - glb_Aai[rowa]; in PCApply_BJKOKKOSKERNELS()
228 int n = glb_Aai[rowa + 1] - glb_Aai[rowa]; in PCApply_BJKOKKOSKERNELS()
229 const PetscInt *aj = glb_Aaj + glb_Aai[rowa]; // global index in PCApply_BJKOKKOSKERNELS()
230 const PetscScalar *aa = glb_Aaa + glb_Aai[rowa]; in PCApply_BJKOKKOSKERNELS()
/petsc/include/petsc/private/
H A Dpcbjkokkosimpl.h69 …_BJKOKKOSKERNELS(PC, const PetscScalar *, PetscScalar *, const PetscInt *glb_Aai, const PetscInt *…