Lines Matching refs:nrecvs

75   PetscMPIInt    nrecvs, tag, done, i;  in PetscCommBuildTwoSided_Ibarrier()  local
92 nrecvs = 0; in PetscCommBuildTwoSided_Ibarrier()
108 nrecvs++; in PetscCommBuildTwoSided_Ibarrier()
123 *nfrom = nrecvs; in PetscCommBuildTwoSided_Ibarrier()
135 PetscMPIInt size, rank, *iflags, nrecvs, tag, *franks, i, flg; in PetscCommBuildTwoSided_Allreduce() local
157 nrecvs = iflags[rank]; in PetscCommBuildTwoSided_Allreduce()
160 PetscCall(PetscMalloc(nrecvs * count * unitbytes, &fdata)); in PetscCommBuildTwoSided_Allreduce()
162 PetscCall(PetscMalloc2(nto + nrecvs, &reqs, nto + nrecvs, &statuses)); in PetscCommBuildTwoSided_Allreduce()
163 sendreqs = PetscSafePointerPlusOffset(reqs, nrecvs); in PetscCommBuildTwoSided_Allreduce()
164 …for (i = 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((void *)(fdata + count * unitbytes * i), coun… in PetscCommBuildTwoSided_Allreduce()
166 PetscCallMPI(MPI_Waitall(nto + nrecvs, reqs, statuses)); in PetscCommBuildTwoSided_Allreduce()
167 PetscCall(PetscMalloc1(nrecvs, &franks)); in PetscCommBuildTwoSided_Allreduce()
168 for (i = 0; i < nrecvs; i++) franks[i] = statuses[i].MPI_SOURCE; in PetscCommBuildTwoSided_Allreduce()
172 *nfrom = nrecvs; in PetscCommBuildTwoSided_Allreduce()
181 PetscMPIInt size, *iflags, nrecvs, tag, *franks, i, flg; in PetscCommBuildTwoSided_RedScatter() local
201 PetscCallMPI(MPI_Reduce_scatter_block(iflags, &nrecvs, 1, MPI_INT, MPI_SUM, comm)); in PetscCommBuildTwoSided_RedScatter()
204 PetscCall(PetscMalloc(nrecvs * count * unitbytes, &fdata)); in PetscCommBuildTwoSided_RedScatter()
206 PetscCall(PetscMalloc2(nto + nrecvs, &reqs, nto + nrecvs, &statuses)); in PetscCommBuildTwoSided_RedScatter()
207 sendreqs = reqs + nrecvs; in PetscCommBuildTwoSided_RedScatter()
208 …for (i = 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((void *)(fdata + count * unitbytes * i), coun… in PetscCommBuildTwoSided_RedScatter()
210 PetscCallMPI(MPI_Waitall(nto + nrecvs, reqs, statuses)); in PetscCommBuildTwoSided_RedScatter()
211 PetscCall(PetscMalloc1(nrecvs, &franks)); in PetscCommBuildTwoSided_RedScatter()
212 for (i = 0; i < nrecvs; i++) franks[i] = statuses[i].MPI_SOURCE; in PetscCommBuildTwoSided_RedScatter()
216 *nfrom = nrecvs; in PetscCommBuildTwoSided_RedScatter()
330 PetscMPIInt nrecvs, tag, *tags, done, i; in PetscCommBuildTwoSidedFReq_Ibarrier() local
360 nrecvs = 0; in PetscCommBuildTwoSidedFReq_Ibarrier()
379 nrecvs++; in PetscCommBuildTwoSidedFReq_Ibarrier()
393 *nfrom = nrecvs; in PetscCommBuildTwoSidedFReq_Ibarrier()