6319   MatCOOStruct_MPIAIJ *coo = *(MatCOOStruct_MPIAIJ **)data;  in MatCOOStructDestroy_MPIAIJ()  local
6322   PetscCall(PetscSFDestroy(&coo->sf));  in MatCOOStructDestroy_MPIAIJ()
6323   PetscCall(PetscFree(coo->Aperm1));  in MatCOOStructDestroy_MPIAIJ()
6324   PetscCall(PetscFree(coo->Bperm1));  in MatCOOStructDestroy_MPIAIJ()
6325   PetscCall(PetscFree(coo->Ajmap1));  in MatCOOStructDestroy_MPIAIJ()
6326   PetscCall(PetscFree(coo->Bjmap1));  in MatCOOStructDestroy_MPIAIJ()
6327   PetscCall(PetscFree(coo->Aimap2));  in MatCOOStructDestroy_MPIAIJ()
6328   PetscCall(PetscFree(coo->Bimap2));  in MatCOOStructDestroy_MPIAIJ()
6329   PetscCall(PetscFree(coo->Aperm2));  in MatCOOStructDestroy_MPIAIJ()
6330   PetscCall(PetscFree(coo->Bperm2));  in MatCOOStructDestroy_MPIAIJ()
6331   PetscCall(PetscFree(coo->Ajmap2));  in MatCOOStructDestroy_MPIAIJ()
6332   PetscCall(PetscFree(coo->Bjmap2));  in MatCOOStructDestroy_MPIAIJ()
6333   PetscCall(PetscFree(coo->Cperm1));  in MatCOOStructDestroy_MPIAIJ()
6334   PetscCall(PetscFree2(coo->sendbuf, coo->recvbuf));  in MatCOOStructDestroy_MPIAIJ()
6335   PetscCall(PetscFree(coo));  in MatCOOStructDestroy_MPIAIJ()
6347   MatCOOStruct_MPIAIJ *coo;  in MatSetPreallocationCOO_MPIAIJ()  local
6671   PetscCall(PetscMalloc1(1, &coo));  in MatSetPreallocationCOO_MPIAIJ()
6672   coo->n       = coo_n;  in MatSetPreallocationCOO_MPIAIJ()
6673   coo->sf      = sf2;  in MatSetPreallocationCOO_MPIAIJ()
6674   coo->sendlen = nleaves;  in MatSetPreallocationCOO_MPIAIJ()
6675   coo->recvlen = nroots;  in MatSetPreallocationCOO_MPIAIJ()
6676   coo->Annz    = Annz;  in MatSetPreallocationCOO_MPIAIJ()
6677   coo->Bnnz    = Bnnz;  in MatSetPreallocationCOO_MPIAIJ()
6678   coo->Annz2   = Annz2;  in MatSetPreallocationCOO_MPIAIJ()
6679   coo->Bnnz2   = Bnnz2;  in MatSetPreallocationCOO_MPIAIJ()
6680   coo->Atot1   = Atot1;  in MatSetPreallocationCOO_MPIAIJ()
6681   coo->Atot2   = Atot2;  in MatSetPreallocationCOO_MPIAIJ()
6682   coo->Btot1   = Btot1;  in MatSetPreallocationCOO_MPIAIJ()
6683   coo->Btot2   = Btot2;  in MatSetPreallocationCOO_MPIAIJ()
6684   coo->Ajmap1  = Ajmap1;  in MatSetPreallocationCOO_MPIAIJ()
6685   coo->Aperm1  = Aperm1;  in MatSetPreallocationCOO_MPIAIJ()
6686   coo->Bjmap1  = Bjmap1;  in MatSetPreallocationCOO_MPIAIJ()
6687   coo->Bperm1  = Bperm1;  in MatSetPreallocationCOO_MPIAIJ()
6688   coo->Aimap2  = Aimap2;  in MatSetPreallocationCOO_MPIAIJ()
6689   coo->Ajmap2  = Ajmap2;  in MatSetPreallocationCOO_MPIAIJ()
6690   coo->Aperm2  = Aperm2;  in MatSetPreallocationCOO_MPIAIJ()
6691   coo->Bimap2  = Bimap2;  in MatSetPreallocationCOO_MPIAIJ()
6692   coo->Bjmap2  = Bjmap2;  in MatSetPreallocationCOO_MPIAIJ()
6693   coo->Bperm2  = Bperm2;  in MatSetPreallocationCOO_MPIAIJ()
6694   coo->Cperm1  = Cperm1;  in MatSetPreallocationCOO_MPIAIJ()
6696   PetscCall(PetscMalloc2(coo->sendlen, &coo->sendbuf, coo->recvlen, &coo->recvbuf));  in MatSetPreallocationCOO_MPIAIJ()
6698   PetscCall(PetscContainerSetPointer(container, coo));  in MatSetPreallocationCOO_MPIAIJ()
6716   MatCOOStruct_MPIAIJ *coo;  in MatSetValuesCOO_MPIAIJ()  local
6721   PetscCall(PetscContainerGetPointer(container, &coo));  in MatSetValuesCOO_MPIAIJ()
6722   sendbuf = coo->sendbuf;  in MatSetValuesCOO_MPIAIJ()
6723   recvbuf = coo->recvbuf;  in MatSetValuesCOO_MPIAIJ()
6724   Ajmap1  = coo->Ajmap1;  in MatSetValuesCOO_MPIAIJ()
6725   Ajmap2  = coo->Ajmap2;  in MatSetValuesCOO_MPIAIJ()
6726   Aimap2  = coo->Aimap2;  in MatSetValuesCOO_MPIAIJ()
6727   Bjmap1  = coo->Bjmap1;  in MatSetValuesCOO_MPIAIJ()
6728   Bjmap2  = coo->Bjmap2;  in MatSetValuesCOO_MPIAIJ()
6729   Bimap2  = coo->Bimap2;  in MatSetValuesCOO_MPIAIJ()
6730   Aperm1  = coo->Aperm1;  in MatSetValuesCOO_MPIAIJ()
6731   Aperm2  = coo->Aperm2;  in MatSetValuesCOO_MPIAIJ()
6732   Bperm1  = coo->Bperm1;  in MatSetValuesCOO_MPIAIJ()
6733   Bperm2  = coo->Bperm2;  in MatSetValuesCOO_MPIAIJ()
6734   Cperm1  = coo->Cperm1;  in MatSetValuesCOO_MPIAIJ()
6740   for (PetscCount i = 0; i < coo->sendlen; i++) sendbuf[i] = v[Cperm1[i]];  in MatSetValuesCOO_MPIAIJ()
6743 …PetscCall(PetscSFReduceWithMemTypeBegin(coo->sf, MPIU_SCALAR, PETSC_MEMTYPE_HOST, sendbuf, PETSC_M…  in MatSetValuesCOO_MPIAIJ()
6745 …for (PetscCount i = 0; i < coo->Annz; i++) { /* All nonzeros in A are either zero'ed or added with…  in MatSetValuesCOO_MPIAIJ()
6750   for (PetscCount i = 0; i < coo->Bnnz; i++) {  in MatSetValuesCOO_MPIAIJ()
6755   PetscCall(PetscSFReduceEnd(coo->sf, MPIU_SCALAR, sendbuf, recvbuf, MPI_REPLACE));  in MatSetValuesCOO_MPIAIJ()
6758   for (PetscCount i = 0; i < coo->Annz2; i++) {  in MatSetValuesCOO_MPIAIJ()
6761   for (PetscCount i = 0; i < coo->Bnnz2; i++) {  in MatSetValuesCOO_MPIAIJ()