| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfbasic.c | 232 PetscCall(PetscMalloc1(bas->itotal, &bas->irootloc)); in PetscSFSetUp_Basic()
234 …for (PetscMPIInt i = bas->ndiranks; i < bas->niranks; i++) PetscCallMPI(MPIU_Irecv(bas->irootloc +… in PetscSFSetUp_Basic()
241 … PetscCall(PetscArraycpy(bas->irootloc + bas->ioffset[0], sf->rremote + sf->roffset[i], npoints)); in PetscSFSetUp_Basic()
266 PetscCall(PetscFree(bas->irootloc)); in PetscSFReset_Basic()
449 …scMPIInt *niranks, const PetscMPIInt **iranks, const PetscInt **ioffset, const PetscInt **irootloc) in PetscSFGetLeafRanks_Basic() argument
457 if (irootloc) *irootloc = bas->irootloc; in PetscSFGetLeafRanks_Basic()
473 const PetscInt *roffset, *rmine, *rremote, *ioffset, *irootloc; in PetscSFCreateEmbeddedRootSF_Basic() local
548 …PetscCall(PetscSFGetRootInfo_Basic(sf, &niranks, &ndiranks, &iranks, &ioffset, &irootloc)); /* Get… in PetscSFCreateEmbeddedRootSF_Basic()
553 PetscCall(PetscMalloc1(ioffset[niranks], &bas->irootloc)); in PetscSFCreateEmbeddedRootSF_Basic()
560 if (rootdata[irootloc[j]]) { in PetscSFCreateEmbeddedRootSF_Basic()
[all …]
|
| H A D | sfbasic.h | 11 …PetscInt *irootloc; /* Incoming roots referenced by ranks starting at ioffset[rank] *…
49 if (rootloc) *rootloc = bas->irootloc; in PetscSFGetRootInfo_Basic()
|
| H A D | sfpack.c | 1324 if (bas->rootbuflen[0]) bas->rootstart[0] = bas->irootloc[0]; in PetscSFSetUpPackFields()
1325 if (bas->rootbuflen[1]) bas->rootstart[1] = bas->irootloc[bas->ioffset[bas->ndiranks]]; in PetscSFSetUpPackFields()
1328 if (bas->irootloc[i] != bas->rootstart[0] + i) { in PetscSFSetUpPackFields()
1334 if (bas->irootloc[i] != bas->rootstart[1] + j) { in PetscSFSetUpPackFields()
1340 …ig[0]) PetscCall(PetscSFCreatePackOpt(bas->ndiranks, bas->ioffset, bas->irootloc, &bas->rootpackop… in PetscSFSetUpPackFields()
1341 …ackOpt(bas->niranks - bas->ndiranks, bas->ioffset + bas->ndiranks, bas->irootloc, &bas->rootpackop… in PetscSFSetUpPackFields()
1348 …ig[0] && !ismulti) PetscCall(PetscCheckDupsInt(bas->rootbuflen[0], bas->irootloc, &bas->rootdups[0… in PetscSFSetUpPackFields()
1349 …ig[1] && !ismulti) PetscCall(PetscCheckDupsInt(bas->rootbuflen[1], bas->irootloc + bas->ioffset[ba… in PetscSFSetUpPackFields()
|
| H A D | sfpack.h | 389 *indices = bas->irootloc + offset; in PetscSFLinkGetRootPackOptAndIndices()
405 …, PETSC_MEMTYPE_DEVICE, bas->irootloc_d[scope], PETSC_MEMTYPE_HOST, bas->irootloc + offset, size)); in PetscSFLinkGetRootPackOptAndIndices()
|
| /petsc/src/vec/is/sf/impls/basic/allgatherv/ |
| H A D | sfallgatherv.c | 82 PetscCall(PetscFree(dat->irootloc)); in PetscSFReset_Allgatherv()
357 …scMPIInt *niranks, const PetscMPIInt **iranks, const PetscInt **ioffset, const PetscInt **irootloc) in PetscSFGetLeafRanks_Allgatherv() argument
393 if (irootloc) { in PetscSFGetLeafRanks_Allgatherv()
394 if (!dat->irootloc) { in PetscSFGetLeafRanks_Allgatherv()
395 PetscCall(PetscMalloc1(sf->nleaves, &dat->irootloc)); in PetscSFGetLeafRanks_Allgatherv()
397 for (PetscInt j = 0; j < sf->nroots; j++) dat->irootloc[i * sf->nroots + j] = j; in PetscSFGetLeafRanks_Allgatherv()
400 *irootloc = dat->irootloc; in PetscSFGetLeafRanks_Allgatherv()
|
| /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 | 149 PetscIntKokkosView irootloc; // plans to put E (i.e., Bd, Bo) into rootBuf member
384 const PetscInt *ioffset, *irootloc, *roffset, *rmine; in MatMPIAIJKokkosReduceBegin() local
392 …PetscCall(PetscSFGetLeafRanks(ownerSF, &niranks, &iranks, &ioffset, &irootloc)); // get leaf rank… in MatMPIAIJKokkosReduceBegin()
472 …PetscCall(PetscSortIntWithPermutation(recvRowCnt, irootloc, recvRowPerm)); // irootloc[] (owned by… in MatMPIAIJKokkosReduceBegin()
486 PetscInt curRowIdx = irootloc[recvRowPerm[iter]]; in MatMPIAIJKokkosReduceBegin()
489 …while (iter + dupRows < recvRowCnt && irootloc[recvRowPerm[iter + dupRows]] == curRowIdx) dupRows+… in MatMPIAIJKokkosReduceBegin()
555 PetscInt curRowIdx = irootloc[recvRowPerm[iter]]; // current row idx in MatMPIAIJKokkosReduceBegin()
559 …while (iter + dupRows < recvRowCnt && irootloc[recvRowPerm[iter + dupRows]] == curRowIdx) dupRows+… in MatMPIAIJKokkosReduceBegin()
793 const PetscInt *ioffset, *irootloc, *roffset; in MatMPIAIJKokkosBcastBegin() local
799 …PetscCall(PetscSFGetLeafRanks(ownerSF, &niranks, &iranks, &ioffset, &irootloc)); // get leaf ranks… in MatMPIAIJKokkosBcastBegin()
[all …]
|
| /petsc/src/vec/is/sf/interface/ftn-custom/ |
| H A D | zsf.c | 57 const PetscInt *irootloc = 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_()
|
| /petsc/src/vec/is/sf/impls/basic/alltoall/ |
| H A D | sfalltoall.c | 154 PetscCall(PetscMalloc1(nroots, &bas->irootloc)); in PetscSFCreateEmbeddedRootSF_Alltoall()
174 bas->irootloc[i] = roots[i]; in PetscSFCreateEmbeddedRootSF_Alltoall()
|
| /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/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()
|
| /petsc/src/vec/is/sf/interface/ |
| H A D | sf.c | 993 …PIInt *niranks, const PetscMPIInt *iranks[], const PetscInt *ioffset[], const PetscInt *irootloc[]) in PetscSFGetLeafRanks() argument
999 PetscUseTypeMethod(sf, GetLeafRanks, niranks, iranks, ioffset, irootloc); in PetscSFGetLeafRanks()
|