10   Mat_MPIAIJ *aij = (Mat_MPIAIJ *)mat->data;  in MatSetUpMultiply_MPIAIJ()  local
11   Mat_SeqAIJ *B   = (Mat_SeqAIJ *)aij->B->data;  in MatSetUpMultiply_MPIAIJ()
25   if (!aij->garray) {  in MatSetUpMultiply_MPIAIJ()
26     PetscCheck(aij->B, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Missing B mat");  in MatSetUpMultiply_MPIAIJ()
29     PetscCall(PetscHMapICreateWithSize(aij->B->rmap->n, &gid1_lid1));  in MatSetUpMultiply_MPIAIJ()
30     for (i = 0; i < aij->B->rmap->n; i++) {  in MatSetUpMultiply_MPIAIJ()
55     for (i = 0; i < aij->B->rmap->n; i++) {  in MatSetUpMultiply_MPIAIJ()
63     PetscCall(PetscLayoutDestroy(&aij->B->cmap));  in MatSetUpMultiply_MPIAIJ()
64 …PetscCall(PetscLayoutCreateFromSizes(PetscObjectComm((PetscObject)aij->B), ec, ec, 1, &aij->B->cma…  in MatSetUpMultiply_MPIAIJ()
70     for (i = 0; i < aij->B->rmap->n; i++) {  in MatSetUpMultiply_MPIAIJ()
88     for (i = 0; i < aij->B->rmap->n; i++) {  in MatSetUpMultiply_MPIAIJ()
91     PetscCall(PetscLayoutDestroy(&aij->B->cmap));  in MatSetUpMultiply_MPIAIJ()
92 …PetscCall(PetscLayoutCreateFromSizes(PetscObjectComm((PetscObject)aij->B), ec, ec, 1, &aij->B->cma…  in MatSetUpMultiply_MPIAIJ()
96     garray = aij->garray;  in MatSetUpMultiply_MPIAIJ()
99   if (!aij->lvec) {  in MatSetUpMultiply_MPIAIJ()
100     PetscCheck(aij->B, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Missing B mat");  in MatSetUpMultiply_MPIAIJ()
101     PetscCall(MatCreateVecs(aij->B, &aij->lvec, NULL));  in MatSetUpMultiply_MPIAIJ()
103   PetscCall(VecGetSize(aij->lvec, &ec));  in MatSetUpMultiply_MPIAIJ()
114   PetscCall(VecScatterDestroy(&aij->Mvctx));  in MatSetUpMultiply_MPIAIJ()
115   PetscCall(VecScatterCreate(gvec, from, aij->lvec, to, &aij->Mvctx));  in MatSetUpMultiply_MPIAIJ()
116   PetscCall(VecScatterViewFromOptions(aij->Mvctx, (PetscObject)mat, "-matmult_vecscatter_view"));  in MatSetUpMultiply_MPIAIJ()
117   aij->garray = garray;  in MatSetUpMultiply_MPIAIJ()
135   Mat_MPIAIJ *aij = (Mat_MPIAIJ *)A->data;  in MatDisAssemble_MPIAIJ()  local
136   Mat         B = aij->B, Bnew = NULL;  in MatDisAssemble_MPIAIJ()
140   PetscCall(VecDestroy(&aij->lvec));  in MatDisAssemble_MPIAIJ()
141   if (aij->colmap) {  in MatDisAssemble_MPIAIJ()
143     PetscCall(PetscHMapIDestroy(&aij->colmap));  in MatDisAssemble_MPIAIJ()
145     PetscCall(PetscFree(aij->colmap));  in MatDisAssemble_MPIAIJ()
151 …  PetscInt           i, j, m = B->rmap->n, n = A->cmap->N, col, ct = 0, *garray = aij->garray, *nz;  in MatDisAssemble_MPIAIJ()
193   PetscCall(PetscFree(aij->garray));  in MatDisAssemble_MPIAIJ()
195   aij->B           = Bnew;  in MatDisAssemble_MPIAIJ()