| /petsc/src/ksp/ksp/tutorials/ |
| H A D | ex61f.F90 | 21 integer :: itmp1, itmp2, ioffset, i local
25 ioffset = 0
27 ibeg(i) = ioffset + 1
28 iend(i) = ioffset + (itmp1 + 1)
29 ioffset = iend(i)
32 ibeg(i) = ioffset + 1
36 iend(i) = ioffset + itmp1
37 ioffset = iend(i)
|
| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfbasic.c | 214 PetscCall(PetscMalloc2(bas->niranks, &bas->iranks, bas->niranks + 1, &bas->ioffset)); in PetscSFSetUp_Basic()
215 bas->ioffset[0] = 0; in PetscSFSetUp_Basic()
218 bas->ioffset[i + 1] = bas->ioffset[i] + rlengths[i]; in PetscSFSetUp_Basic()
223 bas->ioffset[i + 1] = bas->ioffset[i] + ilengths[i - bas->ndiranks]; in PetscSFSetUp_Basic()
225 bas->itotal = bas->ioffset[bas->niranks]; in PetscSFSetUp_Basic()
234 …s; i++) PetscCallMPI(MPIU_Irecv(bas->irootloc + bas->ioffset[i], bas->ioffset[i + 1] - bas->ioffse… in PetscSFSetUp_Basic()
240 …PetscCheck(npoints == bas->ioffset[1] - bas->ioffset[0], PETSC_COMM_SELF, PETSC_ERR_PLIB, "Disting… in PetscSFSetUp_Basic()
241 … PetscCall(PetscArraycpy(bas->irootloc + bas->ioffset[0], sf->rremote + sf->roffset[i], npoints)); in PetscSFSetUp_Basic()
265 PetscCall(PetscFree2(bas->iranks, bas->ioffset)); in PetscSFReset_Basic()
449 …F sf, PetscMPIInt *niranks, const PetscMPIInt **iranks, const PetscInt **ioffset, const PetscInt *… in PetscSFGetLeafRanks_Basic() argument
[all …]
|
| H A D | sfbasic.h | 10 …PetscInt *ioffset; /* Array of length niranks+1 holding offset in irootloc[] for eac…
48 if (rootoffset) *rootoffset = bas->ioffset; in PetscSFGetRootInfo_Basic()
|
| H A D | sfmpi.c | 23 …size_t size = (bas->ioffset[i + bas->ndiranks + 1] - bas->ioffset[i + bas->ndiranks]) * link->unit… in PetscSFLinkFinishCommunication_Default()
|
| H A D | sfpack.c | 1321 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 …ig[1]) PetscCall(PetscSFCreatePackOpt(bas->niranks - bas->ndiranks, bas->ioffset + bas->ndiranks, … in PetscSFSetUpPackFields()
1349 …ti) PetscCall(PetscCheckDupsInt(bas->rootbuflen[1], bas->irootloc + bas->ioffset[bas->ndiranks], &… in PetscSFSetUpPackFields()
|
| H A D | sfpack.h | 386 offset = (scope == PETSCSF_LOCAL) ? 0 : bas->ioffset[bas->ndiranks]; in PetscSFLinkGetRootPackOptAndIndices()
|
| /petsc/src/vec/is/sf/impls/basic/allgatherv/ |
| H A D | sfallgatherv.c | 81 PetscCall(PetscFree(dat->ioffset)); in PetscSFReset_Allgatherv()
357 …F sf, PetscMPIInt *niranks, const PetscMPIInt **iranks, const PetscInt **ioffset, const PetscInt *… in PetscSFGetLeafRanks_Allgatherv() argument
385 if (ioffset) { in PetscSFGetLeafRanks_Allgatherv()
386 if (!dat->ioffset) { in PetscSFGetLeafRanks_Allgatherv()
387 PetscCall(PetscMalloc1(size + 1, &dat->ioffset)); in PetscSFGetLeafRanks_Allgatherv()
388 for (PetscMPIInt i = 0; i <= size; i++) dat->ioffset[i] = i * sf->nroots; in PetscSFGetLeafRanks_Allgatherv()
390 *ioffset = dat->ioffset; in PetscSFGetLeafRanks_Allgatherv()
|
| /petsc/src/vec/is/sf/interface/ftn-custom/ |
| H A D | zsf.c | 56 const PetscInt *ioffset = NULL; in petscsfgetleafranks_() local
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_()
65 …*ierr = F90Array1dCreate((void *)ioffset, MPIU_INT, 1, *niranks + 1, aioffset PETSC_F90_2PTR_PARAM… in petscsfgetleafranks_()
|
| /petsc/src/vec/is/sf/impls/basic/alltoall/ |
| H A D | sfalltoall.c | 153 PetscCall(PetscMalloc2(nroots, &bas->iranks, nroots + 1, &bas->ioffset)); in PetscSFCreateEmbeddedRootSF_Alltoall()
169 bas->ioffset[0] = 0; in PetscSFCreateEmbeddedRootSF_Alltoall()
173 bas->ioffset[i + 1] = i + 1; in PetscSFCreateEmbeddedRootSF_Alltoall()
|
| /petsc/src/vec/is/sf/tutorials/ |
| H A D | ex1f.F90 | 25 PetscInt, pointer :: ioffset(:), irootloc(:), roffset(:), rmine(:), rremote(:)
114 PetscCallA(PetscSFGetLeafRanks(sf, niranks, iranks, ioffset, irootloc, ierr))
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 384 const PetscInt *ioffset, *irootloc, *roffset, *rmine; 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()
405 … 0; i < niranks; i++) PetscCallMPI(MPIU_Irecv(&recvRowLen[ioffset[i]], ioffset[i + 1] - ioffset[i]… in MatMPIAIJKokkosReduceBegin()
413 for (PetscInt j = ioffset[i]; j < ioffset[i + 1]; j++) rdisp[i + 1] += recvRowLen[j]; in MatMPIAIJKokkosReduceBegin()
793 const PetscInt *ioffset, *irootloc, *roffset; in MatMPIAIJKokkosBcastBegin() local
799 …PetscCall(PetscSFGetLeafRanks(ownerSF, &niranks, &iranks, &ioffset, &irootloc)); // get leaf ranks… in MatMPIAIJKokkosBcastBegin()
806 for (PetscInt j = ioffset[i]; j < ioffset[i + 1]; j++) { in MatMPIAIJKokkosBcastBegin()
834 PetscIntKokkosViewHost rowoffset_h(NoInit("rowoffset_h"), ioffset[niranks] + 1); in MatMPIAIJKokkosBcastBegin()
837 …for (PetscInt i = 0; i < ioffset[niranks]; i++) rowoffset[i + 1] = rowoffset[i] + E_RowLen[irootlo… in MatMPIAIJKokkosBcastBegin()
[all …]
|
| /petsc/src/vec/is/sf/impls/basic/nvshmem/ |
| H A D | sfnvshmem.cu | 117 tmp = bas->ioffset[i + bas->ndiranks] - bas->ioffset[bas->ndiranks]; in PetscSFSetUp_Basic_NVSHMEM()
153 …PetscCallCUDA(cudaMemcpyAsync(bas->ioffset_d, bas->ioffset + bas->ndiranks, (nRemoteLeafRanks + 1)… in PetscSFSetUp_Basic_NVSHMEM()
386 …dstdisp_h = bas->ioffset + bas->ndiranks; /* offsets of the local root buf. Note dstdisp[0] is no… in PetscSFLinkGetDataBegin_NVSHMEM()
533 …srcdisp_h = bas->ioffset + bas->ndiranks; /* offsets of rootbuf. srcdisp[0] is not necessarily zer… in PetscSFLinkPutDataBegin_NVSHMEM()
|
| /petsc/src/dm/impls/plex/adaptors/parmmg/ |
| H A D | parmmgadapt.c | 41 const PetscInt *gV, *ioffset, *irootloc, *roffset, *rmine, *rremote; in DMAdaptMetric_ParMmg_Plex() local
85 PetscCall(PetscSFGetLeafRanks(sf, &niranks, &iranks, &ioffset, &irootloc)); in DMAdaptMetric_ParMmg_Plex()
|
| /petsc/src/vec/is/sf/interface/ |
| H A D | sf.c | 993 …F sf, PetscMPIInt *niranks, const PetscMPIInt *iranks[], const PetscInt *ioffset[], const PetscInt… 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()
|