Lines Matching refs:diagPB
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()
315 PetscCall(MatDestroy(&jac->diagPB)); in PCDestroy_PBJacobi()