Home
last modified time | relevance | path

Searched refs:roffset (Results 1 – 20 of 20) sorted by relevance

/petsc/src/dm/impls/plex/
H A Dplexcheckinterface.c77 const PetscInt *roffset, *rmine, *rremote; in SortByRemote_Private() local
81 PetscCall(PetscSFGetRootRanks(sf, &nranks, &ranks, &roffset, &rmine, &rremote)); in SortByRemote_Private()
82 nleaves = roffset[nranks]; in SortByRemote_Private()
87 o = roffset[r]; in SortByRemote_Private()
88 n = roffset[r + 1] - o; in SortByRemote_Private()
101 const PetscInt *roffset; in GetRecursiveConeCoordinatesPerRank_Private() local
106 PetscCall(PetscSFGetRootRanks(sf, &nranks, &ranks, &roffset, NULL, NULL)); in GetRecursiveConeCoordinatesPerRank_Private()
109 o = roffset[r]; in GetRecursiveConeCoordinatesPerRank_Private()
110 n = roffset[r + 1] - o; in GetRecursiveConeCoordinatesPerRank_Private()
171 const PetscInt *mine, *roffset, *rmine, *rremote; in DMPlexCheckInterfaceCones() local
[all …]
H A Dplexinterpolate.c779 const PetscInt *roffset = NULL, *rmine = NULL, *rremote = NULL; in SortRmineRremoteByRemote_Private() local
783 PetscCall(PetscSFGetRootRanks(sf, &nranks, &ranks, &roffset, &rmine, &rremote)); in SortRmineRremoteByRemote_Private()
784 nleaves = roffset[nranks]; in SortRmineRremoteByRemote_Private()
789 o = roffset[r]; in SortRmineRremoteByRemote_Private()
790 n = roffset[r + 1] - o; in SortRmineRremoteByRemote_Private()
804 const PetscInt *roffset; in DMPlexOrientInterface_Internal() local
877 PetscCall(PetscSFGetRootRanks(sf, &nranks, &ranks, &roffset, NULL, NULL)); in DMPlexOrientInterface_Internal()
907 rS = roffset[r]; in DMPlexOrientInterface_Internal()
908 rN = roffset[r + 1] - rS; in DMPlexOrientInterface_Internal()
/petsc/src/vec/is/sf/interface/ftn-custom/
H A Dzsf.c72 const PetscInt *roffset = NULL; in petscsfgetrootranks_() local
76 *ierr = PetscSFGetRootRanks(*sf, nranks, &ranks, &roffset, &rmine, &rremote); in petscsfgetrootranks_()
80 …*ierr = F90Array1dCreate((void *)roffset, MPIU_INT, 1, *nranks + 1, aroffset PETSC_F90_2PTR_PARAM(… in petscsfgetrootranks_()
82 …*ierr = F90Array1dCreate((void *)rmine, MPIU_INT, 1, roffset[*nranks], armine PETSC_F90_2PTR_PARAM… in petscsfgetrootranks_()
84 …*ierr = F90Array1dCreate((void *)rremote, MPIU_INT, 1, roffset[*nranks], arremote PETSC_F90_2PTR_P… in petscsfgetrootranks_()
/petsc/src/vec/is/sf/impls/basic/
H A Dsfbasic.c201 …for (PetscMPIInt i = 0; i < sf->nranks; i++) rlengths[i] = sf->roffset[i + 1] - sf->roffset[i]; /*… in PetscSFSetUp_Basic()
236 PetscInt npoints = sf->roffset[i + 1] - sf->roffset[i]; in PetscSFSetUp_Basic()
241 … PetscCall(PetscArraycpy(bas->irootloc + bas->ioffset[0], sf->rremote + sf->roffset[i], npoints)); in PetscSFSetUp_Basic()
244 …PetscCallMPI(MPIU_Isend(sf->rremote + sf->roffset[i], npoints, MPIU_INT, sf->ranks[i], tag, comm, … in PetscSFSetUp_Basic()
473 const PetscInt *roffset, *rmine, *rremote, *ioffset, *irootloc; in PetscSFCreateEmbeddedRootSF_Basic() local
494 …PetscCall(PetscSFGetLeafInfo_Basic(sf, &nranks, &ndranks, &ranks, &roffset, &rmine, &rremote)); /*… in PetscSFCreateEmbeddedRootSF_Basic()
498 for (j = roffset[i]; j < roffset[i + 1]; j++) { in PetscSFCreateEmbeddedRootSF_Basic()
519 for (j = roffset[i]; j < roffset[i + 1]; j++) { in PetscSFCreateEmbeddedRootSF_Basic()
541 esf->roffset = esf_roffset; in PetscSFCreateEmbeddedRootSF_Basic()
H A Dsfbasic.h59 if (leafoffset) *leafoffset = sf->roffset; in PetscSFGetLeafInfo_Basic()
H A Dsfmpi.c29 … size_t size = (sf->roffset[i + sf->ndranks + 1] - sf->roffset[i + sf->ndranks]) * link->unitbytes; in PetscSFLinkFinishCommunication_Default()
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 …if (!sf->leafcontig[1]) PetscCall(PetscSFCreatePackOpt(sf->nranks - sf->ndranks, sf->roffset + sf-… in PetscSFSetUpPackFields()
1347 …!ismulti) PetscCall(PetscCheckDupsInt(sf->leafbuflen[1], sf->rmine + sf->roffset[sf->ndranks], &sf… in PetscSFSetUpPackFields()
H A Dsfpack.h428 offset = (scope == PETSCSF_LOCAL) ? 0 : sf->roffset[sf->ndranks]; in PetscSFLinkGetLeafPackOptAndIndices()
/petsc/src/vec/is/sf/impls/basic/alltoall/
H A Dsfalltoall.c140 …PetscCall(PetscMalloc4(nleaves, &esf->ranks, nleaves + 1, &esf->roffset, nleaves, &esf->rmine, nle… in PetscSFCreateEmbeddedRootSF_Alltoall()
143 esf->roffset[0] = 0; in PetscSFCreateEmbeddedRootSF_Alltoall()
146 esf->roffset[i + 1] = i + 1; in PetscSFCreateEmbeddedRootSF_Alltoall()
/petsc/src/vec/is/sf/tutorials/
H A Dex1f.F9025 PetscInt, pointer :: ioffset(:), irootloc(:), roffset(:), rmine(:), rremote(:)
115 PetscCallA(PetscSFGetRootRanks(sf, nranks, ranks, roffset, rmine, rremote, ierr))
/petsc/src/vec/is/sf/impls/basic/allgatherv/
H A Dsfallgatherv.c329 …cSF sf, PetscMPIInt *nranks, const PetscMPIInt **ranks, const PetscInt **roffset, const PetscInt *… in PetscSFGetRootRanks_Allgatherv() argument
339 …PetscCall(PetscMalloc4(size, &sf->ranks, size + 1, &sf->roffset, sf->nleaves, &sf->rmine, sf->nlea… in PetscSFGetRootRanks_Allgatherv()
341 PetscCall(PetscArraycpy(sf->roffset, range, size + 1)); in PetscSFGetRootRanks_Allgatherv()
350 if (roffset) *roffset = sf->roffset; in PetscSFGetRootRanks_Allgatherv()
/petsc/src/vec/is/sf/interface/
H A Dsf.c128 PetscCall(PetscFree4(sf->ranks, sf->roffset, sf->rmine, sf->rremote)); in PetscSFReset()
924 …r, "[%d] %d: %" PetscInt_FMT " edges\n", rank, sf->ranks[i], sf->roffset[i + 1] - sf->roffset[i])); in PetscSFView()
925 …for (j = sf->roffset[i]; j < sf->roffset[i + 1]; j++) PetscCall(PetscViewerASCIISynchronizedPrintf… in PetscSFView()
957 …cSF sf, PetscMPIInt *nranks, const PetscMPIInt *ranks[], const PetscInt *roffset[], const PetscInt… in PetscSFGetRootRanks() argument
963 PetscUseTypeMethod(sf, GetRootRanks, nranks, ranks, roffset, rmine, rremote); in PetscSFGetRootRanks()
968 if (roffset) *roffset = sf->roffset; in PetscSFGetRootRanks()
1050 …PetscCall(PetscMalloc4(sf->nranks, &sf->ranks, sf->nranks + 1, &sf->roffset, sf->nleaves, &sf->rmi… in PetscSFSetUpRanks()
1101 sf->roffset[0] = 0; in PetscSFSetUpRanks()
1104 sf->roffset[i + 1] = sf->roffset[i] + rcount[i]; in PetscSFSetUpRanks()
1119 sf->rmine[sf->roffset[irank] + rcount[irank]] = sf->mine ? sf->mine[i] : i; in PetscSFSetUpRanks()
[all …]
/petsc/include/
H A Dpetscsf.h203 …cSF sf, PetscMPIInt *nranks, const PetscMPIInt *ranks[], const PetscInt *roffset[], const PetscInt… in PetscSFGetRanks() argument
205 return PetscSFGetRootRanks(sf, nranks, ranks, roffset, rmine, rremote); in PetscSFGetRanks()
/petsc/src/vec/is/sf/impls/window/
H A Dsfwindow.c75 const PetscInt *roffset; in PetscSFWindowGetDataTypes() local
91 PetscCall(PetscSFGetRootRanks(sf, &nranks, NULL, &roffset, NULL, NULL)); in PetscSFWindowGetDataTypes()
99 PetscCall(PetscMPIIntCast(roffset[i + 1] - roffset[i], &rcount)); in PetscSFWindowGetDataTypes()
101 rmine = sf->rmine + sf->roffset[i]; in PetscSFWindowGetDataTypes()
102 rremote = sf->rremote + sf->roffset[i]; in PetscSFWindowGetDataTypes()
106 PetscCall(PetscMPIIntCast(sf->rmine[sf->roffset[i] + j], &rmine[j])); in PetscSFWindowGetDataTypes()
107 PetscCall(PetscMPIIntCast(sf->rremote[sf->roffset[i] + j], &rremote[j])); in PetscSFWindowGetDataTypes()
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx384 const PetscInt *ioffset, *irootloc, *roffset, *rmine; in MatMPIAIJKokkosReduceBegin() local
393 …PetscCall(PetscSFGetRootRanks(ownerSF, &nranks, &ranks, &roffset, &rmine, NULL)); // get root rank… in MatMPIAIJKokkosReduceBegin()
397 PetscInt sendRowCnt = roffset[nranks], recvRowCnt = ioffset[niranks]; in MatMPIAIJKokkosReduceBegin()
406 …= 0; i < nranks; i++) PetscCallMPI(MPIU_Isend(&sendRowLen[roffset[i]], roffset[i + 1] - roffset[i]… in MatMPIAIJKokkosReduceBegin()
432 for (PetscInt j = roffset[i]; j < roffset[i + 1]; j++) count += E_RowLen[rmine[j]]; in MatMPIAIJKokkosReduceBegin()
447 for (PetscInt k = 0; k < roffset[nranks]; k++) { in MatMPIAIJKokkosReduceBegin()
793 const PetscInt *ioffset, *irootloc, *roffset; in MatMPIAIJKokkosBcastBegin() local
800 PetscCall(PetscSFGetRootRanks(ownerSF, &nranks, &ranks, &roffset, NULL, NULL)); // recv info in MatMPIAIJKokkosBcastBegin()
823 …for (PetscInt j = Fi[roffset[i]]; j < Fi[roffset[i + 1]]; j++) { // I will receive rows [roffset[i… in MatMPIAIJKokkosBcastBegin()
/petsc/src/dm/impls/plex/adaptors/parmmg/
H A Dparmmgadapt.c41 const PetscInt *gV, *ioffset, *irootloc, *roffset, *rmine, *rremote; in DMAdaptMetric_ParMmg_Plex() local
86 PetscCall(PetscSFGetRootRanks(sf, &nrranks, &rranks, &roffset, &rmine, &rremote)); in DMAdaptMetric_ParMmg_Plex()
88 for (i = roffset[r]; i < roffset[r + 1]; ++i) { in DMAdaptMetric_ParMmg_Plex()
/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu130 …PetscCallCUDA(cudaMemcpyAsync(sf->roffset_d, sf->roffset + sf->ndranks, (nRemoteRootRanks + 1) * s… in PetscSFSetUp_Basic_NVSHMEM()
140 tmp = sf->roffset[i + sf->ndranks] - sf->roffset[sf->ndranks]; in PetscSFSetUp_Basic_NVSHMEM()
368 …dstdisp_h = sf->roffset + sf->ndranks; /* offsets of the local leaf buf. Note dstdisp[0] is not n… in PetscSFLinkGetDataBegin_NVSHMEM()
546 srcdisp_h = sf->roffset + sf->ndranks; /* offsets of leafbuf */ in PetscSFLinkPutDataBegin_NVSHMEM()
/petsc/src/dm/dt/dualspace/impls/sum/
H A Ddualspacesum.c392 PetscInt *roffset; in PetscDualSpaceSumCreateMappings() local
411 PetscCall(PetscCalloc1(pEnd - pStart, &roffset)); in PetscDualSpaceSumCreateMappings()
434 PetscCall(ISCreateStride(PETSC_COMM_SELF, subNb, roffset[0], rstride, &is_row)); in PetscDualSpaceSumCreateMappings()
435 roffset[0] += interleave_basis ? 1 : subNb; in PetscDualSpaceSumCreateMappings()
449 … for (PetscInt k = 0; k < subdof; k++) rows[suboff + k] = off + roffset[p - pStart] + k * stride; in PetscDualSpaceSumCreateMappings()
450 roffset[p - pStart] += interleave_basis ? 1 : subdof; in PetscDualSpaceSumCreateMappings()
482 PetscCall(PetscFree(roffset)); in PetscDualSpaceSumCreateMappings()
/petsc/include/petsc/private/
H A Dsfimpl.h77 PetscInt *roffset; /* Array of length nranks+1, offset in rmine/rremote for each rank */ member
/petsc/src/vec/vec/utils/
H A Dvscat.c161 for (i = 0; i < sf->roffset[sf->nranks]; i++) sf->rremote[i] = PETSC_INT_MIN; in VecScatterRemap_Internal()