Lines Matching refs:nrecvs
33 …sages(MPI_Comm comm, const PetscMPIInt iflags[], const PetscMPIInt ilengths[], PetscMPIInt *nrecvs) in PetscGatherNumberOfMessages() argument
55 *nrecvs = recv_buf[rank]; in PetscGatherNumberOfMessages()
90 PetscErrorCode PetscGatherMessageLengths(MPI_Comm comm, PetscMPIInt nsends, PetscMPIInt nrecvs, con… in PetscGatherMessageLengths() argument
102 PetscCall(PetscMalloc2(nrecvs + nsends, &r_waits, nrecvs + nsends, &w_status)); in PetscGatherMessageLengths()
103 s_waits = PetscSafePointerPlusOffset(r_waits, nrecvs); in PetscGatherMessageLengths()
106 PetscCall(PetscMalloc1(nrecvs, olengths)); in PetscGatherMessageLengths()
107 …for (i = 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((*olengths) + i, 1, MPI_INT, MPI_ANY_SOURCE, … in PetscGatherMessageLengths()
118 if (nrecvs + nsends) PetscCallMPI(MPI_Waitall(nrecvs + nsends, r_waits, w_status)); in PetscGatherMessageLengths()
121 PetscCall(PetscMalloc1(nrecvs, onodes)); in PetscGatherMessageLengths()
122 for (i = 0; i < nrecvs; ++i) { in PetscGatherMessageLengths()
139 …_Private(MPI_Comm comm, const PetscMPIInt iflags[], const PetscInt ilengths[], PetscMPIInt *nrecvs) in PetscGatherNumberOfMessages_Private() argument
161 *nrecvs = recv_buf[rank]; in PetscGatherNumberOfMessages_Private()
168 …herMessageLengths_Private(MPI_Comm comm, PetscMPIInt nsends, PetscMPIInt nrecvs, const PetscInt il… in PetscGatherMessageLengths_Private() argument
180 PetscCall(PetscMalloc2(nrecvs + nsends, &r_waits, nrecvs + nsends, &w_status)); in PetscGatherMessageLengths_Private()
181 s_waits = PetscSafePointerPlusOffset(r_waits, nrecvs); in PetscGatherMessageLengths_Private()
184 PetscCall(PetscMalloc1(nrecvs, olengths)); in PetscGatherMessageLengths_Private()
185 …for (i = 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((*olengths) + i, 1, MPIU_INT, MPI_ANY_SOURCE,… in PetscGatherMessageLengths_Private()
196 if (nrecvs + nsends) PetscCallMPI(MPI_Waitall(nrecvs + nsends, r_waits, w_status)); in PetscGatherMessageLengths_Private()
199 PetscCall(PetscMalloc1(nrecvs, onodes)); in PetscGatherMessageLengths_Private()
200 for (i = 0; i < nrecvs; ++i) { in PetscGatherMessageLengths_Private()
239 PetscErrorCode PetscGatherMessageLengths2(MPI_Comm comm, PetscMPIInt nsends, PetscMPIInt nrecvs, co… in PetscGatherMessageLengths2() argument
250 …PetscCall(PetscMalloc4(nrecvs + nsends, &r_waits, 2 * nrecvs, &buf_r, 2 * nsends, &buf_s, nrecvs +… in PetscGatherMessageLengths2()
251 s_waits = PetscSafePointerPlusOffset(r_waits, nrecvs); in PetscGatherMessageLengths2()
254 PetscCall(PetscMalloc1(nrecvs, olengths1)); in PetscGatherMessageLengths2()
255 PetscCall(PetscMalloc1(nrecvs, olengths2)); in PetscGatherMessageLengths2()
256 for (i = 0; i < nrecvs; i++) { in PetscGatherMessageLengths2()
274 if (nrecvs + nsends) PetscCallMPI(MPI_Waitall(nrecvs + nsends, r_waits, w_status)); in PetscGatherMessageLengths2()
277 PetscCall(PetscMalloc1(nrecvs, onodes)); in PetscGatherMessageLengths2()
278 for (i = 0; i < nrecvs; ++i) { in PetscGatherMessageLengths2()
293 PetscErrorCode PetscPostIrecvInt(MPI_Comm comm, PetscMPIInt tag, PetscMPIInt nrecvs, const PetscMPI… in PetscPostIrecvInt() argument
300 for (i = 0; i < nrecvs; i++) len += olengths[i]; /* each message length */ in PetscPostIrecvInt()
303 PetscCall(PetscMalloc1(nrecvs + 1, &rbuf_t)); in PetscPostIrecvInt()
305 for (i = 1; i < nrecvs; ++i) rbuf_t[i] = rbuf_t[i - 1] + olengths[i - 1]; in PetscPostIrecvInt()
308 PetscCall(PetscMalloc1(nrecvs, &r_waits_t)); in PetscPostIrecvInt()
309 …for (i = 0; i < nrecvs; ++i) PetscCallMPI(MPIU_Irecv(rbuf_t[i], olengths[i], MPIU_INT, onodes[i], … in PetscPostIrecvInt()
316 PetscErrorCode PetscPostIrecvScalar(MPI_Comm comm, PetscMPIInt tag, PetscMPIInt nrecvs, const Petsc… in PetscPostIrecvScalar() argument
325 for (i = 0; i < nrecvs; i++) len += olengths[i]; /* each message length */ in PetscPostIrecvScalar()
328 PetscCall(PetscMalloc1(nrecvs + 1, &rbuf_t)); in PetscPostIrecvScalar()
330 for (i = 1; i < nrecvs; ++i) rbuf_t[i] = rbuf_t[i - 1] + olengths[i - 1]; in PetscPostIrecvScalar()
333 PetscCall(PetscMalloc1(nrecvs, &r_waits_t)); in PetscPostIrecvScalar()
334 …for (i = 0; i < nrecvs; ++i) PetscCallMPI(MPIU_Irecv(rbuf_t[i], olengths[i], MPIU_SCALAR, onodes[i… in PetscPostIrecvScalar()