Searched refs:diagPB (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/ksp/pc/impls/pbjacobi/ |
| H A D | pbjacobi.c | 248 PETSC_INTERN PetscErrorCode PCSetUp_PBJacobi_Host(PC pc, Mat diagPB) in PCSetUp_PBJacobi_Host() argument
251 Mat A = diagPB ? diagPB : pc->pmat; in PCSetUp_PBJacobi_Host()
269 Mat diagPB = NULL; in PCSetUp_PBJacobi() local
277 …if (flg) PetscUseTypeMethod(pc->pmat, getblockdiagonal, &diagPB); // diagPB's reference count is i… in PCSetUp_PBJacobi()
282 …if (!isCuda && diagPB) PetscCall(PetscObjectTypeCompareAny((PetscObject)diagPB, &isCuda, MATSEQAIJ… in PCSetUp_PBJacobi()
287 …if (!isKok && diagPB) PetscCall(PetscObjectTypeCompareAny((PetscObject)diagPB, &isKok, MATSEQAIJKO… in PCSetUp_PBJacobi()
291 if (isCuda) PetscCall(PCSetUp_PBJacobi_CUDA(pc, diagPB)); in PCSetUp_PBJacobi()
296 PetscCall(PCSetUp_PBJacobi_Kokkos(pc, diagPB)); in PCSetUp_PBJacobi()
300 PetscCall(PCSetUp_PBJacobi_Host(pc, diagPB)); in PCSetUp_PBJacobi()
302 …PetscCall(MatDestroy(&diagPB)); // since we don't need it anymore, we don't stash it in PC_PBJacobi in PCSetUp_PBJacobi()
[all …]
|
| H A D | pbjacobi.h | 11 …Mat diagPB; /* the matrix made of the diagonal blocks if some shell smatrix provided… member
|
| /petsc/src/ksp/pc/impls/pbjacobi/kokkos/ |
| H A D | pbjacobi_kok.kokkos.cxx | 82 PETSC_INTERN PetscErrorCode PCSetUp_PBJacobi_Kokkos(PC pc, Mat diagPB) in PCSetUp_PBJacobi_Kokkos() argument
88 …PetscCall(PCSetUp_PBJacobi_Host(pc, diagPB)); /* Compute the inverse on host now. Might worth doin… in PCSetUp_PBJacobi_Kokkos()
|