Lines Matching refs:nrqr

62   PetscMPIInt      size, rank, tag1, tag2, *w2, *w1, nrqs, nrqr, *pa;  in MatIncreaseOverlap_MPIBAIJ_Once()  local
135 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr)); in MatIncreaseOverlap_MPIBAIJ_Once()
136 PetscCall(PetscGatherMessageLengths(comm, nrqs, nrqr, w1, &onodes1, &olengths1)); in MatIncreaseOverlap_MPIBAIJ_Once()
139 PetscCall(PetscPostIrecvInt(comm, tag1, nrqr, onodes1, olengths1, &rbuf, &r_waits1)); in MatIncreaseOverlap_MPIBAIJ_Once()
231 PetscCallMPI(MPI_Waitall(nrqr, r_waits1, MPI_STATUSES_IGNORE)); in MatIncreaseOverlap_MPIBAIJ_Once()
238 PetscCall(PetscMalloc1(nrqr, &xdata)); in MatIncreaseOverlap_MPIBAIJ_Once()
239 PetscCall(PetscMalloc1(nrqr, &isz1)); in MatIncreaseOverlap_MPIBAIJ_Once()
240 PetscCall(MatIncreaseOverlap_MPIBAIJ_Receive(C, nrqr, rbuf, xdata, isz1)); in MatIncreaseOverlap_MPIBAIJ_Once()
252 for (PetscMPIInt i = 0; i < nrqr; ++i) { in MatIncreaseOverlap_MPIBAIJ_Once()
258 PetscCall(PetscGatherMessageLengths(comm, nrqr, nrqs, rw1, &onodes2, &olengths2)); in MatIncreaseOverlap_MPIBAIJ_Once()
265 PetscCall(PetscMalloc1(nrqr, &s_waits2)); in MatIncreaseOverlap_MPIBAIJ_Once()
266 for (PetscMPIInt i = 0; i < nrqr; ++i) { in MatIncreaseOverlap_MPIBAIJ_Once()
299 PetscCallMPI(MPI_Waitall(nrqr, s_waits2, MPI_STATUSES_IGNORE)); in MatIncreaseOverlap_MPIBAIJ_Once()
404 static PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Receive(Mat C, PetscInt nrqr, PetscInt **rbuf, Pet… in MatIncreaseOverlap_MPIBAIJ_Receive() argument
425 for (i = 0, ct = 0, total_sz = 0; i < nrqr; ++i) { in MatIncreaseOverlap_MPIBAIJ_Receive()
435 if (nrqr) { in MatIncreaseOverlap_MPIBAIJ_Receive()
440 PetscCall(PetscArrayzero(isz1, nrqr)); in MatIncreaseOverlap_MPIBAIJ_Receive()
443 for (i = 0; i < nrqr; i++) { /* for easch mesg from proc i */ in MatIncreaseOverlap_MPIBAIJ_Receive()
516 if (i + 1 < nrqr) xdata[i + 1] = xdata[i] + ct2; in MatIncreaseOverlap_MPIBAIJ_Receive()
621 …PetscMPIInt rank, size, tag0, tag2, tag3, tag4, *w1, *w2, *w3, *w4, nrqr, nrqs = 0, *req_sour… local
702 nrqr = smat_i->nrqr;
730 nrqr = smat_i->nrqr;
807 PetscCall(PetscGatherNumberOfMessages(comm, w2, w1, &nrqr));
808 PetscCall(PetscGatherMessageLengths(comm, nrqs, nrqr, w1, &onodes1, &olengths1));
812 PetscCall(PetscPostIrecvInt(comm, tag0, nrqr, onodes1, olengths1, &rbuf1, &r_waits1));
888 PetscCall(PetscMalloc1(nrqr, &s_waits2));
889 PetscCall(PetscMalloc3(nrqr, &sbuf2, nrqr, &req_size, nrqr, &req_source1));
891 PetscCallMPI(MPI_Waitall(nrqr, r_waits1, MPI_STATUSES_IGNORE));
892 for (PetscMPIInt i = 0; i < nrqr; ++i) {
931 PetscCallMPI(MPI_Waitall(nrqr, s_waits2, MPI_STATUSES_IGNORE));
936 PetscCall(PetscMalloc1(nrqr, &sbuf_aj));
938 for (PetscMPIInt i = 0; i < nrqr; i++) jcnt += req_size[i];
939 if (nrqr) PetscCall(PetscMalloc1(jcnt, &sbuf_aj[0]));
940 for (PetscMPIInt i = 1; i < nrqr; i++) sbuf_aj[i] = sbuf_aj[i - 1] + req_size[i - 1];
942 PetscCall(PetscMalloc1(nrqr, &s_waits3));
944 for (PetscMPIInt i = 0; i < nrqr; i++) {
1131 PetscCallMPI(MPI_Waitall(nrqr, s_waits3, MPI_STATUSES_IGNORE));
1158 smat_i->nrqr = nrqr;
1197 smat_i->nrqr = nrqr;
1240 PetscCall(PetscMalloc1(nrqr, &sbuf_aa));
1242 for (PetscMPIInt i = 0; i < nrqr; i++) jcnt += req_size[i];
1243 if (nrqr) PetscCall(PetscMalloc1(jcnt * bs2, &sbuf_aa[0]));
1244 for (PetscMPIInt i = 1; i < nrqr; i++) sbuf_aa[i] = sbuf_aa[i - 1] + req_size[i - 1] * bs2;
1246 PetscCall(PetscMalloc1(nrqr, &s_waits4));
1248 for (PetscMPIInt i = 0; i < nrqr; i++) {
1513 PetscCallMPI(MPI_Waitall(nrqr, s_waits4, MPI_STATUSES_IGNORE));