Lines Matching refs:count
73 static PetscErrorCode PetscCommBuildTwoSided_Ibarrier(MPI_Comm comm, PetscMPIInt count, MPI_Datatyp… in PetscCommBuildTwoSided_Ibarrier() argument
88 …for (i = 0; i < nto; i++) PetscCallMPI(MPI_Issend((void *)(tdata + count * unitbytes * i), count, … in PetscCommBuildTwoSided_Ibarrier()
90 PetscCall(PetscSegBufferCreate(unitbytes, 4 * count, &segdata)); in PetscCommBuildTwoSided_Ibarrier()
105 PetscCall(PetscSegBufferGet(segdata, count, &buf)); in PetscCommBuildTwoSided_Ibarrier()
107 PetscCallMPI(MPI_Recv(buf, count, dtype, status.MPI_SOURCE, tag, comm, MPI_STATUS_IGNORE)); in PetscCommBuildTwoSided_Ibarrier()
133 static PetscErrorCode PetscCommBuildTwoSided_Allreduce(MPI_Comm comm, PetscMPIInt count, MPI_Dataty… in PetscCommBuildTwoSided_Allreduce() argument
160 PetscCall(PetscMalloc(nrecvs * count * unitbytes, &fdata)); in PetscCommBuildTwoSided_Allreduce()
164 … 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((void *)(fdata + count * unitbytes * i), count, dtype… in PetscCommBuildTwoSided_Allreduce()
165 …for (i = 0; i < nto; i++) PetscCallMPI(MPIU_Isend((void *)(tdata + count * unitbytes * i), count, … in PetscCommBuildTwoSided_Allreduce()
179 static PetscErrorCode PetscCommBuildTwoSided_RedScatter(MPI_Comm comm, PetscMPIInt count, MPI_Datat… in PetscCommBuildTwoSided_RedScatter() argument
204 PetscCall(PetscMalloc(nrecvs * count * unitbytes, &fdata)); in PetscCommBuildTwoSided_RedScatter()
208 … 0; i < nrecvs; i++) PetscCallMPI(MPIU_Irecv((void *)(fdata + count * unitbytes * i), count, dtype… in PetscCommBuildTwoSided_RedScatter()
209 …for (i = 0; i < nto; i++) PetscCallMPI(MPIU_Isend((void *)(tdata + count * unitbytes * i), count, … in PetscCommBuildTwoSided_RedScatter()
255 PetscErrorCode PetscCommBuildTwoSided(MPI_Comm comm, PetscMPIInt count, MPI_Datatype dtype, PetscMP… in PetscCommBuildTwoSided() argument
267 …PetscCall(PetscCommBuildTwoSided_Ibarrier(comm, count, dtype, nto, toranks, todata, nfrom, fromran… in PetscCommBuildTwoSided()
273 …PetscCall(PetscCommBuildTwoSided_Allreduce(comm, count, dtype, nto, toranks, todata, nfrom, fromra… in PetscCommBuildTwoSided()
277 …PetscCall(PetscCommBuildTwoSided_RedScatter(comm, count, dtype, nto, toranks, todata, nfrom, fromr… in PetscCommBuildTwoSided()
289 static PetscErrorCode PetscCommBuildTwoSidedFReq_Reference(MPI_Comm comm, PetscMPIInt count, MPI_Da… in PetscCommBuildTwoSidedFReq_Reference() argument
301 …PetscCall(PetscCommBuildTwoSided(comm, count, dtype, nto, toranks, todata, nfrom, fromranks, fromd… in PetscCommBuildTwoSidedFReq_Reference()
311 …PetscCall((*send)(comm, tag, i, toranks[i], ((char *)todata) + count * unitbytes * i, sendreq + i … in PetscCommBuildTwoSidedFReq_Reference()
314 void *header = (*(char **)fromdata) + count * unitbytes * i; in PetscCommBuildTwoSidedFReq_Reference()
328 static PetscErrorCode PetscCommBuildTwoSidedFReq_Ibarrier(MPI_Comm comm, PetscMPIInt count, MPI_Dat… in PetscCommBuildTwoSidedFReq_Ibarrier() argument
347 …for (i = 0; i < nto; i++) PetscCallMPI(MPI_Issend((void *)(tdata + count * unitbytes * i), count, … in PetscCommBuildTwoSidedFReq_Ibarrier()
353 …PetscCall((*send)(comm, tags, i, toranks[i], tdata + count * unitbytes * i, usendreqs + i * ntags,… in PetscCommBuildTwoSidedFReq_Ibarrier()
357 PetscCall(PetscSegBufferCreate(unitbytes, 4 * count, &segdata)); in PetscCommBuildTwoSidedFReq_Ibarrier()
373 PetscCall(PetscSegBufferGet(segdata, count, &buf)); in PetscCommBuildTwoSidedFReq_Ibarrier()
375 PetscCallMPI(MPI_Recv(buf, count, dtype, status.MPI_SOURCE, tag, comm, MPI_STATUS_IGNORE)); in PetscCommBuildTwoSidedFReq_Ibarrier()
440 PetscErrorCode PetscCommBuildTwoSidedF(MPI_Comm comm, PetscMPIInt count, MPI_Datatype dtype, PetscM… in PetscCommBuildTwoSidedF() argument
445 …PetscCall(PetscCommBuildTwoSidedFReq(comm, count, dtype, nto, toranks, todata, nfrom, fromranks, f… in PetscCommBuildTwoSidedF()
487 PetscErrorCode PetscCommBuildTwoSidedFReq(MPI_Comm comm, PetscMPIInt count, MPI_Datatype dtype, Pet… in PetscCommBuildTwoSidedFReq() argument
515 …PetscCall((*f)(comm, count, dtype, nto, toranks, todata, nfrom, fromranks, fromdata, ntags, toreqs… in PetscCommBuildTwoSidedFReq()