Home
last modified time | relevance | path

Searched refs:ndiranks (Results 1 – 7 of 7) sorted by relevance

/petsc/src/vec/is/sf/impls/basic/alltoall/
H A Dsfalltoall.c93 PetscMPIInt nroots, *roots, nleaves, *leaves, rank, ndiranks, ndranks; in PetscSFCreateEmbeddedRootSF_Alltoall() local
156 ndiranks = 0; in PetscSFCreateEmbeddedRootSF_Alltoall()
160 ndiranks = 1; in PetscSFCreateEmbeddedRootSF_Alltoall()
168 bas->ndiranks = ndiranks; in PetscSFCreateEmbeddedRootSF_Alltoall()
179 bas->nrootreqs = bas->niranks - bas->ndiranks; in PetscSFCreateEmbeddedRootSF_Alltoall()
/petsc/src/vec/is/sf/impls/basic/
H A Dsfbasic.c212 bas->ndiranks = sf->ndranks; in PetscSFSetUp_Basic()
213 bas->niranks = bas->ndiranks + niranks; in PetscSFSetUp_Basic()
216 for (PetscMPIInt i = 0; i < bas->ndiranks; i++) { in PetscSFSetUp_Basic()
220 …PetscCheck(bas->ndiranks <= 1 && (bas->ndiranks != 1 || bas->iranks[0] == rank), PETSC_COMM_SELF, … in PetscSFSetUp_Basic()
221 for (PetscMPIInt i = bas->ndiranks; i < bas->niranks; i++) { in PetscSFSetUp_Basic()
222 bas->iranks[i] = iranks[i - bas->ndiranks]; in PetscSFSetUp_Basic()
223 bas->ioffset[i + 1] = bas->ioffset[i] + ilengths[i - bas->ndiranks]; in PetscSFSetUp_Basic()
231 nRemoteLeafRanks = bas->niranks - bas->ndiranks; in PetscSFSetUp_Basic()
234 …ndiranks; i < bas->niranks; i++) PetscCallMPI(MPIU_Irecv(bas->irootloc + bas->ioffset[i], bas->iof… in PetscSFSetUp_Basic()
471 PetscMPIInt *esf_ranks, nranks, ndranks, niranks, esf_nranks, esf_ndranks, ndiranks; in PetscSFCreateEmbeddedRootSF_Basic() local
[all …]
H A Dsfbasic.h7 …PetscMPIInt ndiranks; /* Number of incoming ranks (ranks accessing my roots) in disting…
46 if (ndrootranks) *ndrootranks = bas->ndiranks; in PetscSFGetRootInfo_Basic()
H A Dsfmpi.c23 …size_t size = (bas->ioffset[i + bas->ndiranks + 1] - bas->ioffset[i + bas->ndiranks]) * link->unit… in PetscSFLinkFinishCommunication_Default()
24 … bytes) with MPI tag %10d ... ", rank, rootaction, bas->iranks[i + bas->ndiranks], size, link->tag… in PetscSFLinkFinishCommunication_Default()
H A Dsfpack.c1321 bas->rootbuflen[0] = bas->ioffset[bas->ndiranks]; in PetscSFSetUpPackFields()
1322 bas->rootbuflen[1] = bas->ioffset[bas->niranks] - bas->ioffset[bas->ndiranks]; in PetscSFSetUpPackFields()
1325 if (bas->rootbuflen[1]) bas->rootstart[1] = bas->irootloc[bas->ioffset[bas->ndiranks]]; in PetscSFSetUpPackFields()
1327 for (i = 0; i < bas->ioffset[bas->ndiranks]; i++) { in PetscSFSetUpPackFields()
1333 for (i = bas->ioffset[bas->ndiranks], j = 0; i < bas->ioffset[bas->niranks]; i++, j++) { in PetscSFSetUpPackFields()
1340 …if (!bas->rootcontig[0]) PetscCall(PetscSFCreatePackOpt(bas->ndiranks, bas->ioffset, bas->irootloc… in PetscSFSetUpPackFields()
1341 …g[1]) PetscCall(PetscSFCreatePackOpt(bas->niranks - bas->ndiranks, bas->ioffset + bas->ndiranks, b… in PetscSFSetUpPackFields()
1349 …PetscCheckDupsInt(bas->rootbuflen[1], bas->irootloc + bas->ioffset[bas->ndiranks], &bas->rootdups[… in PetscSFSetUpPackFields()
H A Dsfpack.h386 offset = (scope == PETSCSF_LOCAL) ? 0 : bas->ioffset[bas->ndiranks]; in PetscSFLinkGetRootPackOptAndIndices()
/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu89 nRemoteLeafRanks = bas->niranks - bas->ndiranks; in PetscSFSetUp_Basic_NVSHMEM()
112 …Ranks; i++) PetscCallMPI(MPI_Send(&i, 1, MPIU_INT, bas->iranks[i + bas->ndiranks], tag, comm)); … in PetscSFSetUp_Basic_NVSHMEM()
117 tmp = bas->ioffset[i + bas->ndiranks] - bas->ioffset[bas->ndiranks]; in PetscSFSetUp_Basic_NVSHMEM()
118 …PetscCallMPI(MPI_Send(&tmp, 1, MPIU_INT, bas->iranks[i + bas->ndiranks], tag, comm)); /* Roots sen… in PetscSFSetUp_Basic_NVSHMEM()
134 …lMPI(MPIU_Irecv(&bas->leafsigdisp[i], 1, MPIU_INT, bas->iranks[i + bas->ndiranks], tag, comm, &roo… in PetscSFSetUp_Basic_NVSHMEM()
138 …lMPI(MPIU_Irecv(&bas->leafbufdisp[i], 1, MPIU_INT, bas->iranks[i + bas->ndiranks], tag, comm, &roo… in PetscSFSetUp_Basic_NVSHMEM()
152 …PetscCallCUDA(cudaMemcpyAsync(bas->iranks_d, bas->iranks + bas->ndiranks, nRemoteLeafRanks * sizeo… in PetscSFSetUp_Basic_NVSHMEM()
153 …PetscCallCUDA(cudaMemcpyAsync(bas->ioffset_d, bas->ioffset + bas->ndiranks, (nRemoteLeafRanks + 1)… in PetscSFSetUp_Basic_NVSHMEM()
380 srcranks_h = bas->iranks + bas->ndiranks; /* my (remote) root ranks */ in PetscSFLinkGetDataBegin_NVSHMEM()
386 …dstdisp_h = bas->ioffset + bas->ndiranks; /* offsets of the local root buf. Note dstdisp[0] is no… in PetscSFLinkGetDataBegin_NVSHMEM()
[all …]