Lines Matching refs:subA
7976 PetscErrorCode MatSelectVariableBlockSizes(Mat subA, Mat A, IS isrow) in MatSelectVariableBlockSizes() argument
7996 …I(MPIU_Allreduce(MPI_IN_PLACE, &flg, 1, MPI_C_BOOL, MPI_LAND, PetscObjectComm((PetscObject)subA))); in MatSelectVariableBlockSizes()
8017 subA->nblocks = Nb; in MatSelectVariableBlockSizes()
8018 PetscCall(PetscFree(subA->bsizes)); in MatSelectVariableBlockSizes()
8019 PetscCall(PetscMalloc1(subA->nblocks, &subA->bsizes)); in MatSelectVariableBlockSizes()
8022 if (sb < subA->nblocks) subA->bsizes[sb] = 0; in MatSelectVariableBlockSizes()
8028 ++subA->bsizes[sb]; in MatSelectVariableBlockSizes()
8034 if (sb < subA->nblocks && subA->bsizes[sb]) ++sb; in MatSelectVariableBlockSizes()
8036 …etscCheck(sb == subA->nblocks, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of blocks %" Petsc… in MatSelectVariableBlockSizes()
8038 PetscCall(MatGetLocalSize(subA, &nlocal, NULL)); in MatSelectVariableBlockSizes()
8039 …subA->nblocks >= 0 && subA->nblocks <= nlocal, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Number of loca… in MatSelectVariableBlockSizes()
8040 for (PetscInt i = 0; i < subA->nblocks; i++) ncnt += subA->bsizes[i]; in MatSelectVariableBlockSizes()