Lines Matching refs:iflags
135 PetscMPIInt size, rank, *iflags, nrecvs, tag, *franks, i, flg; in PetscCommBuildTwoSided_Allreduce() local
148 if (!counter->iflags) { in PetscCommBuildTwoSided_Allreduce()
149 PetscCall(PetscCalloc1(size, &counter->iflags)); in PetscCommBuildTwoSided_Allreduce()
150 iflags = counter->iflags; in PetscCommBuildTwoSided_Allreduce()
152 iflags = counter->iflags; in PetscCommBuildTwoSided_Allreduce()
153 PetscCall(PetscArrayzero(iflags, size)); in PetscCommBuildTwoSided_Allreduce()
155 for (i = 0; i < nto; i++) iflags[toranks[i]] = 1; in PetscCommBuildTwoSided_Allreduce()
156 PetscCallMPI(MPIU_Allreduce(MPI_IN_PLACE, iflags, size, MPI_INT, MPI_SUM, comm)); in PetscCommBuildTwoSided_Allreduce()
157 nrecvs = iflags[rank]; in PetscCommBuildTwoSided_Allreduce()
181 PetscMPIInt size, *iflags, nrecvs, tag, *franks, i, flg; in PetscCommBuildTwoSided_RedScatter() local
193 if (!counter->iflags) { in PetscCommBuildTwoSided_RedScatter()
194 PetscCall(PetscCalloc1(size, &counter->iflags)); in PetscCommBuildTwoSided_RedScatter()
195 iflags = counter->iflags; in PetscCommBuildTwoSided_RedScatter()
197 iflags = counter->iflags; in PetscCommBuildTwoSided_RedScatter()
198 PetscCall(PetscArrayzero(iflags, size)); in PetscCommBuildTwoSided_RedScatter()
200 for (i = 0; i < nto; i++) iflags[toranks[i]] = 1; in PetscCommBuildTwoSided_RedScatter()
201 PetscCallMPI(MPI_Reduce_scatter_block(iflags, &nrecvs, 1, MPI_INT, MPI_SUM, comm)); in PetscCommBuildTwoSided_RedScatter()