Lines Matching refs:merge
1591 MatMergeSeqsToMPI *merge; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ() local
1621 merge = ap->merge; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1625 coi = merge->coi; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1626 coj = merge->coj; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1628 bi = merge->bi; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1629 bj = merge->bj; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1630 owners = merge->rowmap->range; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1693 buf_ri = merge->buf_ri; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1694 buf_rj = merge->buf_rj; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1695 len_s = merge->len_s; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1697 …PetscCall(PetscPostIrecvScalar(comm, taga, merge->nrecv, merge->id_r, merge->len_r, &abuf_r, &r_wa… in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1699 PetscCall(PetscMalloc2(merge->nsend, &s_waits, size, &status)); in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1702 i = merge->owners_co[proc]; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1706 if (merge->nrecv) PetscCallMPI(MPI_Waitall(merge->nrecv, r_waits, status)); in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1707 if (merge->nsend) PetscCallMPI(MPI_Waitall(merge->nsend, s_waits, status)); in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1714 PetscCall(PetscMalloc3(merge->nrecv, &buf_ri_k, merge->nrecv, &nextrow, merge->nrecv, &nextci)); in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1715 for (k = 0; k < merge->nrecv; k++) { in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1728 for (k = 0; k < merge->nrecv; k++) { /* k-th received message */ in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ()
1775 MatMergeSeqsToMPI *merge; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ() local
1860 PetscCall(PetscNew(&merge)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1861 PetscCall(PetscLayoutCreate(comm, &merge->rowmap)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1863 merge->rowmap->n = pn; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1864 merge->rowmap->bs = 1; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1866 PetscCall(PetscLayoutSetUp(merge->rowmap)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1867 owners = merge->rowmap->range; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1871 PetscCall(PetscCalloc1(size, &merge->len_s)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1873 len_s = merge->len_s; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1874 merge->nsend = 0; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1890 merge->nsend++; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1897 PetscCall(PetscGatherNumberOfMessages(comm, NULL, len_s, &merge->nrecv)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1898 …PetscCall(PetscGatherMessageLengths2(comm, merge->nsend, merge->nrecv, len_s, len_si, &merge->id_r… in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1902 …PetscCall(PetscPostIrecvInt(comm, tagj, merge->nrecv, merge->id_r, merge->len_r, &buf_rj, &rwaits)… in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1903 PetscCall(PetscMalloc1(merge->nsend, &swaits)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1913 for (i = 0; i < merge->nrecv; i++) { in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1915 PetscCallMPI(MPI_Waitany(merge->nrecv, rwaits, &icompleted, &rstatus)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1918 if (merge->nsend) PetscCallMPI(MPI_Waitall(merge->nsend, swaits, sstatus)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1922 for (k = 0; k < merge->nrecv; k++) { /* k-th received message */ in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1924 for (j = 0; j < merge->len_r[k]; j++) PetscCall(PetscHMapISet(ta, *(Jptr + j) + 1, 1)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1930 PetscCall(PetscPostIrecvInt(comm, tagi, merge->nrecv, merge->id_r, len_ri, &buf_ri, &rwaits)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1955 i = merge->nrecv; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1958 PetscCallMPI(MPI_Waitany(merge->nrecv, rwaits, &icompleted, &rstatus)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1961 if (merge->nsend) PetscCallMPI(MPI_Waitall(merge->nsend, swaits, sstatus)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1978 PetscCall(PetscMalloc3(merge->nrecv, &buf_ri_k, merge->nrecv, &nextrow, merge->nrecv, &nextci)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
1979 for (k = 0; k < merge->nrecv; k++) { in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2002 for (k = 0; k < merge->nrecv; k++) { /* k-th received message */ in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2065 merge->bi = bi; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2066 merge->bj = bj; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2067 merge->coi = coi; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2068 merge->coj = coj; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2069 merge->buf_ri = buf_ri; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2070 merge->buf_rj = buf_rj; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2071 merge->owners_co = owners_co; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()
2076 ap->merge = merge; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ()