Home
last modified time | relevance | path

Searched refs:ndranks (Results 1 – 9 of 9) 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
116 ndranks = 0; in PetscSFCreateEmbeddedRootSF_Alltoall()
120 ndranks = 1; in PetscSFCreateEmbeddedRootSF_Alltoall()
142 esf->ndranks = ndranks; in PetscSFCreateEmbeddedRootSF_Alltoall()
178 esf->nleafreqs = esf->nranks - esf->ndranks; in PetscSFCreateEmbeddedRootSF_Alltoall()
/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu88 nRemoteRootRanks = sf->nranks - sf->ndranks; in PetscSFSetUp_Basic_NVSHMEM()
111 …scCallMPI(MPIU_Irecv(&sf->rootsigdisp[i], 1, MPIU_INT, sf->ranks[i + sf->ndranks], tag, comm, &lea… in PetscSFSetUp_Basic_NVSHMEM()
115 …scCallMPI(MPIU_Irecv(&sf->rootbufdisp[i], 1, MPIU_INT, sf->ranks[i + sf->ndranks], tag, comm, &lea… in PetscSFSetUp_Basic_NVSHMEM()
129 …PetscCallCUDA(cudaMemcpyAsync(sf->ranks_d, sf->ranks + sf->ndranks, nRemoteRootRanks * sizeof(Pets… in PetscSFSetUp_Basic_NVSHMEM()
130 …PetscCallCUDA(cudaMemcpyAsync(sf->roffset_d, sf->roffset + sf->ndranks, (nRemoteRootRanks + 1) * s… in PetscSFSetUp_Basic_NVSHMEM()
135 …moteRootRanks; i++) PetscCallMPI(MPI_Send(&i, 1, MPIU_INT, sf->ranks[i + sf->ndranks], tag, comm)); in PetscSFSetUp_Basic_NVSHMEM()
140 tmp = sf->roffset[i + sf->ndranks] - sf->roffset[sf->ndranks]; in PetscSFSetUp_Basic_NVSHMEM()
141 PetscCallMPI(MPI_Send(&tmp, 1, MPIU_INT, sf->ranks[i + sf->ndranks], tag, comm)); in PetscSFSetUp_Basic_NVSHMEM()
362 srcranks_h = sf->ranks + sf->ndranks; /* my (remote) root ranks */ in PetscSFLinkGetDataBegin_NVSHMEM()
368 …dstdisp_h = sf->roffset + sf->ndranks; /* offsets of the local leaf buf. Note dstdisp[0] is not n… in PetscSFLinkGetDataBegin_NVSHMEM()
[all …]
/petsc/src/vec/is/sf/impls/basic/
H A Dsfbasic.c202 nRemoteRootRanks = sf->nranks - sf->ndranks; in PetscSFSetUp_Basic()
203 …nks, PetscSafePointerPlusOffset(sf->ranks, sf->ndranks), PetscSafePointerPlusOffset(rlengths, sf->… in PetscSFSetUp_Basic()
212 bas->ndiranks = sf->ndranks; in PetscSFSetUp_Basic()
237 if (i < sf->ndranks) { in PetscSFSetUp_Basic()
244 …rremote + sf->roffset[i], npoints, MPIU_INT, sf->ranks[i], tag, comm, &leafreqs[i - sf->ndranks])); in PetscSFSetUp_Basic()
471 PetscMPIInt *esf_ranks, nranks, ndranks, niranks, esf_nranks, esf_ndranks, ndiranks; in PetscSFCreateEmbeddedRootSF_Basic() local
494 …PetscCall(PetscSFGetLeafInfo_Basic(sf, &nranks, &ndranks, &ranks, &roffset, &rmine, &rremote)); /*… in PetscSFCreateEmbeddedRootSF_Basic()
506 if (i < ndranks) esf_ndranks++; in PetscSFCreateEmbeddedRootSF_Basic()
539 esf->ndranks = esf_ndranks; in PetscSFCreateEmbeddedRootSF_Basic()
H A Dsfmpi.c29 … size_t size = (sf->roffset[i + sf->ndranks + 1] - sf->roffset[i + sf->ndranks]) * link->unitbytes; in PetscSFLinkFinishCommunication_Default()
30 …16zu bytes) with MPI tag %10d ... ", rank, leafaction, sf->ranks[i + sf->ndranks], size, link->tag… in PetscSFLinkFinishCommunication_Default()
H A Dsfbasic.h57 if (ndleafranks) *ndleafranks = sf->ndranks; in PetscSFGetLeafInfo_Basic()
H A Dsfpack.c1296 sf->leafbuflen[0] = sf->roffset[sf->ndranks]; in PetscSFSetUpPackFields()
1297 sf->leafbuflen[1] = sf->roffset[sf->nranks] - sf->roffset[sf->ndranks]; in PetscSFSetUpPackFields()
1300 if (sf->leafbuflen[1]) sf->leafstart[1] = sf->rmine[sf->roffset[sf->ndranks]]; in PetscSFSetUpPackFields()
1303 for (i = 0; i < sf->roffset[sf->ndranks]; i++) { /* self */ in PetscSFSetUpPackFields()
1309 for (i = sf->roffset[sf->ndranks], j = 0; i < sf->roffset[sf->nranks]; i++, j++) { /* remote */ in PetscSFSetUpPackFields()
1317 …if (!sf->leafcontig[0]) PetscCall(PetscSFCreatePackOpt(sf->ndranks, sf->roffset, sf->rmine, &sf->l… in PetscSFSetUpPackFields()
1318 …contig[1]) PetscCall(PetscSFCreatePackOpt(sf->nranks - sf->ndranks, sf->roffset + sf->ndranks, sf-… in PetscSFSetUpPackFields()
1347 …tscCall(PetscCheckDupsInt(sf->leafbuflen[1], sf->rmine + sf->roffset[sf->ndranks], &sf->leafdups[1… in PetscSFSetUpPackFields()
H A Dsfpack.h428 offset = (scope == PETSCSF_LOCAL) ? 0 : sf->roffset[sf->ndranks]; in PetscSFLinkGetLeafPackOptAndIndices()
/petsc/src/vec/is/sf/interface/
H A Dsf.c643 …sf->ndranks = 0; /* We do not need to separate out distinguished ranks for patterned graphs to im… in PetscSFSetGraphWithPattern()
1078 for (sf->ndranks = 0, i = sf->nranks; sf->ndranks < i;) { in PetscSFSetUpRanks()
1079 for (i--; sf->ndranks < i; i--) { /* Scan i backward looking for distinguished rank */ in PetscSFSetUpRanks()
1082 …for (; sf->ndranks <= i; sf->ndranks++) { /* Scan sf->ndranks forward looking for non-distinguishe… in PetscSFSetUpRanks()
1083 if (!InList(ranks[sf->ndranks], groupsize, groupranks)) break; in PetscSFSetUpRanks()
1085 if (sf->ndranks < i) { /* Swap ranks[sf->ndranks] with ranks[i] */ in PetscSFSetUpRanks()
1091 ranks[i] = ranks[sf->ndranks]; in PetscSFSetUpRanks()
1092 rcount[i] = rcount[sf->ndranks]; in PetscSFSetUpRanks()
1093 ranks[sf->ndranks] = tmprank; in PetscSFSetUpRanks()
1094 rcount[sf->ndranks] = tmpcount; in PetscSFSetUpRanks()
[all …]
/petsc/include/petsc/private/
H A Dsfimpl.h75 …PetscMPIInt ndranks; /* Number of ranks in distinguished group holding roots connected to my l… member