| /petsc/src/dm/impls/plex/ |
| H A D | plexcheckinterface.c | 123 PetscMPIInt niranks; in PetscSFComputeMultiRootOriginalNumberingByRank_Private() local
130 PetscCall(PetscSFGetRootRanks(imsf, &niranks, NULL, &iroffset, &irmine, NULL)); in PetscSFComputeMultiRootOriginalNumberingByRank_Private()
131 …PetscCheck(nileaves == iroffset[niranks], PETSC_COMM_SELF, PETSC_ERR_PLIB, "nileaves != iroffset[n… in PetscSFComputeMultiRootOriginalNumberingByRank_Private()
136 for (PetscMPIInt r = 0; r < niranks; r++) { in PetscSFComputeMultiRootOriginalNumberingByRank_Private()
175 PetscMPIInt niranks; in DMPlexCheckInterfaceCones() local
213 PetscCall(PetscSFGetRootRanks(imsf, &niranks, &iranks, &iroffset, &irmine, &irremote)); in DMPlexCheckInterfaceCones()
222 …VecByRank_Private((PetscObject)sf, nranks, ranks, sntCoordinatesPerRank, niranks, iranks, &recCoor… in DMPlexCheckInterfaceCones()
246 if (r < niranks) { in DMPlexCheckInterfaceCones()
260 if (r < niranks) { in DMPlexCheckInterfaceCones()
276 for (r = 0; r < niranks; r++) { in DMPlexCheckInterfaceCones()
[all …]
|
| H A D | plexcreate.c | 5677 PetscMPIInt niranks, njranks; in DMGetNeighbors_Plex() local
5687 PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, NULL, NULL)); in DMGetNeighbors_Plex()
5688 PetscCall(PetscMalloc1(njranks + niranks + 1, &data->neighbors)); in DMGetNeighbors_Plex()
5690 PetscCall(PetscArraycpy(data->neighbors + njranks + 1, iranks, niranks)); in DMGetNeighbors_Plex()
5691 n = njranks + niranks; in DMGetNeighbors_Plex()
|
| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfbasic.c | 184 PetscMPIInt rank, niranks, *iranks, tag; in PetscSFSetUp_Basic() local
203 …ranks, sf->ndranks), PetscSafePointerPlusOffset(rlengths, sf->ndranks), &niranks, &iranks, (void *… in PetscSFSetUp_Basic()
209 PetscCall(PetscSortMPIIntWithIntArray(niranks, iranks, ilengths)); in PetscSFSetUp_Basic()
213 bas->niranks = bas->ndiranks + niranks; in PetscSFSetUp_Basic()
214 PetscCall(PetscMalloc2(bas->niranks, &bas->iranks, bas->niranks + 1, &bas->ioffset)); in PetscSFSetUp_Basic()
221 for (PetscMPIInt i = bas->ndiranks; i < bas->niranks; i++) { in PetscSFSetUp_Basic()
225 bas->itotal = bas->ioffset[bas->niranks]; in PetscSFSetUp_Basic()
231 nRemoteLeafRanks = bas->niranks - bas->ndiranks; in PetscSFSetUp_Basic()
234 …for (PetscMPIInt i = bas->ndiranks; i < bas->niranks; i++) PetscCallMPI(MPIU_Irecv(bas->irootloc +… in PetscSFSetUp_Basic()
449 PETSC_INTERN PetscErrorCode PetscSFGetLeafRanks_Basic(PetscSF sf, PetscMPIInt *niranks, const Petsc… in PetscSFGetLeafRanks_Basic() argument
[all …]
|
| H A D | sfbasic.h | 6 PetscMPIInt niranks; /* Number of incoming ranks (ranks accessing my roots) */ \
45 if (nrootranks) *nrootranks = bas->niranks; in PetscSFGetRootInfo_Basic()
|
| H A D | sfpack.c | 1322 bas->rootbuflen[1] = bas->ioffset[bas->niranks] - bas->ioffset[bas->ndiranks]; in PetscSFSetUpPackFields()
1333 for (i = bas->ioffset[bas->ndiranks], j = 0; i < bas->ioffset[bas->niranks]; i++, j++) { in PetscSFSetUpPackFields()
1341 …if (!bas->rootcontig[1]) PetscCall(PetscSFCreatePackOpt(bas->niranks - bas->ndiranks, bas->ioffset… in PetscSFSetUpPackFields()
|
| /petsc/src/vec/is/sf/interface/ftn-custom/ |
| H A D | zsf.c | 53 PETSC_EXTERN void petscsfgetleafranks_(PetscSF *sf, PetscMPIInt *niranks, F90Array1d *airanks, F90A… in petscsfgetleafranks_() argument
59 *ierr = PetscSFGetLeafRanks(*sf, niranks, &iranks, &ioffset, &irootloc); in petscsfgetleafranks_()
61 …*ierr = F90Array1dCreate((void *)irootloc, MPIU_INT, 1, ioffset[*niranks], airootloc PETSC_F90_2PT… in petscsfgetleafranks_()
63 …*ierr = F90Array1dCreate((void *)iranks, MPI_INT, 1, *niranks, airanks PETSC_F90_2PTR_PARAM(pirank… in petscsfgetleafranks_()
65 …*ierr = F90Array1dCreate((void *)ioffset, MPIU_INT, 1, *niranks + 1, aioffset PETSC_F90_2PTR_PARAM… in petscsfgetleafranks_()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 385 PetscMPIInt niranks, nranks; in MatMPIAIJKokkosReduceBegin() local
392 …PetscCall(PetscSFGetLeafRanks(ownerSF, &niranks, &iranks, &ioffset, &irootloc)); // get leaf rank… in MatMPIAIJKokkosReduceBegin()
397 PetscInt sendRowCnt = roffset[nranks], recvRowCnt = ioffset[niranks]; in MatMPIAIJKokkosReduceBegin()
400 …&sendRowLen, recvRowCnt + 1, &recvRowLen, nranks, &sdisp, niranks + 1, &rdisp, nranks + niranks, &… in MatMPIAIJKokkosReduceBegin()
405 …for (PetscInt i = 0; i < niranks; i++) PetscCallMPI(MPIU_Irecv(&recvRowLen[ioffset[i]], ioffset[i … in MatMPIAIJKokkosReduceBegin()
407 PetscCallMPI(MPI_Waitall(nranks + niranks, reqs, MPI_STATUSES_IGNORE)); in MatMPIAIJKokkosReduceBegin()
411 for (PetscInt i = 0; i < niranks; i++) { in MatMPIAIJKokkosReduceBegin()
419 …for (PetscInt i = 0; i < niranks; i++) PetscCallMPI(MPIU_Isend(&rdisp[i], 1, MPIU_INT, iranks[i], … in MatMPIAIJKokkosReduceBegin()
420 PetscCallMPI(MPI_Waitall(nranks + niranks, reqs, MPI_STATUSES_IGNORE)); in MatMPIAIJKokkosReduceBegin()
423 PetscInt nroots = rdisp[niranks]; // roots are nonzeros I will recv in MatMPIAIJKokkosReduceBegin()
[all …]
|
| /petsc/src/vec/is/sf/tutorials/ |
| H A D | ex1f.F90 | 23 PetscMPIInt niranks, nranks
114 PetscCallA(PetscSFGetLeafRanks(sf, niranks, iranks, ioffset, irootloc, ierr))
|
| /petsc/src/vec/is/sf/impls/basic/alltoall/ |
| H A D | sfalltoall.c | 167 bas->niranks = nroots; in PetscSFCreateEmbeddedRootSF_Alltoall()
179 bas->nrootreqs = bas->niranks - bas->ndiranks; in PetscSFCreateEmbeddedRootSF_Alltoall()
|
| /petsc/src/vec/is/sf/impls/basic/allgatherv/ |
| H A D | sfallgatherv.c | 357 PETSC_INTERN PetscErrorCode PetscSFGetLeafRanks_Allgatherv(PetscSF sf, PetscMPIInt *niranks, const … in PetscSFGetLeafRanks_Allgatherv() argument
368 if (niranks) *niranks = size; in PetscSFGetLeafRanks_Allgatherv()
|
| /petsc/src/dm/impls/plex/adaptors/parmmg/ |
| H A D | parmmgadapt.c | 37 PetscMPIInt niranks, nrranks, numNgbRanks; in DMAdaptMetric_ParMmg_Plex() local
81 niranks = nrranks = 0; in DMAdaptMetric_ParMmg_Plex()
85 PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, &ioffset, &irootloc)); in DMAdaptMetric_ParMmg_Plex()
|
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 89 nRemoteLeafRanks = bas->niranks - bas->ndiranks; in PetscSFSetUp_Basic_NVSHMEM()
|
| /petsc/src/vec/is/sf/interface/ |
| H A D | sf.c | 993 PetscErrorCode PetscSFGetLeafRanks(PetscSF sf, PetscMPIInt *niranks, const PetscMPIInt *iranks[], c… in PetscSFGetLeafRanks() argument
999 PetscUseTypeMethod(sf, GetLeafRanks, niranks, iranks, ioffset, irootloc); in PetscSFGetLeafRanks()
|
| /petsc/src/vec/vec/utils/ |
| H A D | vscat.c | 168 …for (i = 0; i < bas->ioffset[bas->niranks]; i++) bas->irootloc[i] = tomap[bas->irootloc[i] * bs] /… in VecScatterRemap_Internal()
|