Lines Matching refs:w1
495 PetscMPIInt *w1, *w2, nrqr, *w3, *w4, *onodes1, *olengths1, *onodes2, *olengths2; in MatIncreaseOverlap_MPIAIJ_Once() local
532 PetscCall(PetscCalloc4(size, &w1, size, &w2, size, &w3, size, &w4)); in MatIncreaseOverlap_MPIAIJ_Once()
545 w1[j] += w4[j]; in MatIncreaseOverlap_MPIAIJ_Once()
553 w1[rank] = 0; /* no mesg sent to intself */ in MatIncreaseOverlap_MPIAIJ_Once()
556 if (w1[i]) { in MatIncreaseOverlap_MPIAIJ_Once()
564 if (w1[i]) { in MatIncreaseOverlap_MPIAIJ_Once()
573 w1[j] += w2[j] + 2 * w3[j]; in MatIncreaseOverlap_MPIAIJ_Once()
574 msz += w1[j]; in MatIncreaseOverlap_MPIAIJ_Once()
578 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr)); in MatIncreaseOverlap_MPIAIJ_Once()
579 PetscCall(PetscGatherMessageLengths(comm, nrqs, nrqr, w1, &onodes1, &olengths1)); in MatIncreaseOverlap_MPIAIJ_Once()
595 ict = w1[j]; in MatIncreaseOverlap_MPIAIJ_Once()
678 PetscCallMPI(MPIU_Isend(outdat[j], w1[j], MPIU_INT, j, tag1, comm, s_waits1 + i)); in MatIncreaseOverlap_MPIAIJ_Once()
705 PetscCall(PetscFree4(w1, w2, w3, w4)); in MatIncreaseOverlap_MPIAIJ_Once()
1222 …PetscMPIInt rank, size, *req_source1, *req_source2, tag1, tag2, tag3, tag4, *w1, *w2, nrqr, nr… in MatCreateSubMatrices_MPIAIJ_SingleIS_Local() local
1278 PetscCall(PetscCalloc2(size, &w1, size, &w2)); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1288 w1[proc]++; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1296 w1[rank] = 0; /* rows owned by self will not be requested */ in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1300 if (w1[proc]) pa[j++] = proc; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1306 w1[pa[i]] += 3; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1307 msz += w1[pa[i]]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1313 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr)); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1317 PetscCall(PetscGatherMessageLengths(comm, nrqs, nrqr, w1, &onodes1, &olengths1)); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1334 iptr += w1[pa[i]]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1366 …for (PetscMPIInt i = 0; i < nrqs; ++i) PetscCallMPI(MPIU_Isend(sbuf1[pa[i]], w1[pa[i]], MPIU_INT, … in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1373 for (PetscMPIInt i = 1; i < nrqs; ++i) rbuf2[i] = rbuf2[i - 1] + w1[pa[i - 1]]; in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1375 …for (PetscMPIInt i = 0; i < nrqs; ++i) PetscCallMPI(MPIU_Irecv(rbuf2[i], w1[pa[i]], MPIU_INT, pa[i… in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
1377 PetscCall(PetscFree2(w1, w2)); in MatCreateSubMatrices_MPIAIJ_SingleIS_Local()
2042 …PetscMPIInt rank, size, tag0, tag2, tag3, tag4, *w1, *w2, *w3, *w4, nrqr, *req_source1 = NULL… in MatCreateSubMatrices_MPIAIJ_Local() local
2158 …PetscCall(PetscCalloc4(size, &w1, size, &w2, size, &w3, size, &w4)); /* mesg size, initialize work… in MatCreateSubMatrices_MPIAIJ_Local()
2177 w1[j] += w4[j]; in MatCreateSubMatrices_MPIAIJ_Local()
2186 w1[rank] = 0; /* no mesg sent to self */ in MatCreateSubMatrices_MPIAIJ_Local()
2189 if (w1[i]) { in MatCreateSubMatrices_MPIAIJ_Local()
2196 if (w1[i]) { in MatCreateSubMatrices_MPIAIJ_Local()
2205 w1[j] += w2[j] + 2 * w3[j]; in MatCreateSubMatrices_MPIAIJ_Local()
2206 msz += w1[j]; in MatCreateSubMatrices_MPIAIJ_Local()
2211 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr)); in MatCreateSubMatrices_MPIAIJ_Local()
2212 PetscCall(PetscGatherMessageLengths(comm, nrqs, nrqr, w1, &onodes1, &olengths1)); in MatCreateSubMatrices_MPIAIJ_Local()
2230 k = w1[j]; in MatCreateSubMatrices_MPIAIJ_Local()
2271 PetscCallMPI(MPIU_Isend(sbuf1[j], w1[j], MPIU_INT, j, tag0, comm, s_waits1 + i)); in MatCreateSubMatrices_MPIAIJ_Local()
2278 for (PetscMPIInt i = 1; i < nrqs; ++i) rbuf2[i] = rbuf2[i - 1] + w1[pa[i - 1]]; in MatCreateSubMatrices_MPIAIJ_Local()
2281 PetscCallMPI(MPIU_Irecv(rbuf2[i], w1[j], MPIU_INT, j, tag2, comm, r_waits2 + i)); in MatCreateSubMatrices_MPIAIJ_Local()
2318 PetscCall(PetscFree4(w1, w2, w3, w4)); in MatCreateSubMatrices_MPIAIJ_Local()