Home
last modified time | relevance | path

Searched refs:diagVPB (Results 1 – 4 of 4) sorted by relevance

/petsc/src/ksp/pc/impls/vpbjacobi/
H A Dvpbjacobi.c208 PETSC_INTERN PetscErrorCode PCSetUp_VPBJacobi_Host(PC pc, Mat diagVPB) in PCSetUp_VPBJacobi_Host() argument
211 Mat A = diagVPB ? diagVPB : pc->pmat; in PCSetUp_VPBJacobi_Host()
244 Mat diagVPB = NULL; in PCSetUp_VPBJacobi() local
252 …if (flg) PetscUseTypeMethod(pc->pmat, getvblockdiagonal, &diagVPB); // diagVPB's reference count i… in PCSetUp_VPBJacobi()
257 …if (!isCuda && diagVPB) PetscCall(PetscObjectTypeCompareAny((PetscObject)diagVPB, &isCuda, MATSEQA… in PCSetUp_VPBJacobi()
262 …if (!isKok && diagVPB) PetscCall(PetscObjectTypeCompareAny((PetscObject)diagVPB, &isKok, MATSEQAIJ… in PCSetUp_VPBJacobi()
266 if (isCuda) PetscCall(PCSetUp_VPBJacobi_CUDA(pc, diagVPB)); in PCSetUp_VPBJacobi()
271 PetscCall(PCSetUp_VPBJacobi_Kokkos(pc, diagVPB)); in PCSetUp_VPBJacobi()
275 PetscCall(PCSetUp_VPBJacobi_Host(pc, diagVPB)); in PCSetUp_VPBJacobi()
277 …PetscCall(MatDestroy(&diagVPB)); // since we don't need it anymore, we don't need to stash it in P… in PCSetUp_VPBJacobi()
[all …]
H A Dvpbjacobi.h11 …Mat diagVPB; /* the matrix made of the diagonal blocks if some shell matrix… member
/petsc/src/ksp/pc/impls/vpbjacobi/kokkos/
H A Dvpbjacobi_kok.kokkos.cxx144 PETSC_INTERN PetscErrorCode PCSetUp_VPBJacobi_Kokkos(PC pc, Mat diagVPB) in PCSetUp_VPBJacobi_Kokkos() argument
186 if (diagVPB) { // If caller provided a matrix made of the diagonal blocks, use it in PCSetUp_VPBJacobi_Kokkos()
187 PetscCall(PetscObjectBaseTypeCompare((PetscObject)diagVPB, MATMPIAIJ, &ismpi)); in PCSetUp_VPBJacobi_Kokkos()
188 A = ismpi ? static_cast<Mat_MPIAIJ *>(diagVPB->data)->A : diagVPB; in PCSetUp_VPBJacobi_Kokkos()
/petsc/src/ksp/pc/impls/vpbjacobi/cuda/
H A Dvpbjacobi_cuda.cu160 PETSC_INTERN PetscErrorCode PCSetUp_VPBJacobi_CUDA(PC pc, Mat diagVPB) in PCSetUp_VPBJacobi_CUDA() argument
168 …PetscCall(PCSetUp_VPBJacobi_Host(pc, diagVPB)); /* Compute the inverse on host now. Might worth do… in PCSetUp_VPBJacobi_CUDA()