Home
last modified time | relevance | path

Searched refs:leafbuflen (Results 1 – 8 of 8) sorted by relevance

/petsc/src/vec/is/sf/impls/basic/allgather/
H A Dsfallgather.c13 sf->leafbuflen[i] = 0; in PetscSFSetUp_Allgather()
23 sf->leafbuflen[PETSCSF_REMOTE] = sf->nleaves; in PetscSFSetUp_Allgather()
75 …->leafmtype_mpi]) PetscCall(PetscSFMalloc(sf, link->leafmtype_mpi, sf->leafbuflen[PETSCSF_REMOTE] … in PetscSFReduceBegin_Allgather()
107 …C_MEMTYPE_HOST, link->leafbuf[PETSCSF_REMOTE][PETSC_MEMTYPE_HOST], sf->leafbuflen[PETSCSF_REMOTE] … in PetscSFBcastToZero_Allgather()
/petsc/src/vec/is/sf/impls/basic/
H A Dsfmpi.c197 if (sf->leafbuflen[i]) { in PetscSFLinkCreate_MPI()
201 …eafbuf_alloc[i][leafmtype]) PetscCall(PetscSFMalloc(sf, leafmtype, sf->leafbuflen[i] * link->unitb… in PetscSFLinkCreate_MPI()
214 …f_alloc[PETSCSF_REMOTE][PETSC_MEMTYPE_HOST]) PetscCall(PetscMalloc(sf->leafbuflen[PETSCSF_REMOTE] … in PetscSFLinkCreate_MPI()
H A Dsfpack.h324 …evice(link->leafmtype) && PetscMemTypeHost(link->leafmtype_mpi) && sf->leafbuflen[PETSCSF_REMOTE])… in PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI()
327 size_t count = sf->leafbuflen[PETSCSF_REMOTE] * link->unitbytes; in PetscSFLinkCopyLeafBufferInCaseNotUseGpuAwareMPI()
346 …buflen[PETSCSF_REMOTE]) || (PetscMemTypeDevice(link->leafmtype) && sf->leafbuflen[PETSCSF_REMOTE])… in PetscSFLinkSyncStreamBeforeCallMPI()
423 *count = sf->leafbuflen[scope]; in PetscSFLinkGetLeafPackOptAndIndices()
445 size = sf->leafbuflen[scope] * sizeof(PetscInt); in PetscSFLinkGetLeafPackOptAndIndices()
H A Dsfpack.c839 …if (PetscMemTypeDevice(link->leafmtype)) PetscCall(PetscLogGpuFlops(sf->leafbuflen[scope] * link->… in PetscSFLinkLogFlopsAfterUnpackLeafData()
842 PetscCall(PetscLogFlops(sf->leafbuflen[scope] * link->bs)); in PetscSFLinkLogFlopsAfterUnpackLeafData()
978 if (sf->leafbuflen[scope]) PetscCall(PetscSFLinkPackLeafData_Private(sf, link, scope, leafdata)); in PetscSFLinkPackLeafData()
1050 …if (sf->leafbuflen[scope] && !link->leafdirect[scope]) PetscCall(PetscSFLinkUnpackLeafData_Private… in PetscSFLinkUnpackLeafData()
1089 PetscInt buflen = sf->leafbuflen[PETSCSF_LOCAL]; in PetscSFLinkScatterLocal()
1296 sf->leafbuflen[0] = sf->roffset[sf->ndranks]; in PetscSFSetUpPackFields()
1297 sf->leafbuflen[1] = sf->roffset[sf->nranks] - sf->roffset[sf->ndranks]; in PetscSFSetUpPackFields()
1299 if (sf->leafbuflen[0]) sf->leafstart[0] = sf->rmine[0]; in PetscSFSetUpPackFields()
1300 if (sf->leafbuflen[1]) sf->leafstart[1] = sf->rmine[sf->roffset[sf->ndranks]]; in PetscSFSetUpPackFields()
1346 …if (!sf->leafcontig[0] && !ismulti) PetscCall(PetscCheckDupsInt(sf->leafbuflen[0], sf->rmine, &sf-… in PetscSFSetUpPackFields()
[all …]
H A Dsfbasic.c37 …if (sf->leafbuflen[PETSCSF_REMOTE] && !link->leafreqsinited[direction][leafmtype_mpi][leafdirect_m… in PetscSFLinkInitMPIRequests_Persistent_Basic()
66 …rbuflen = (direction == PETSCSF_ROOT2LEAF) ? sf->leafbuflen[PETSCSF_REMOTE] : bas->rootbuflen[PETS… in PetscSFLinkStartCommunication_Persistent_Basic()
77 …sbuflen = (direction == PETSCSF_ROOT2LEAF) ? bas->rootbuflen[PETSCSF_REMOTE] : sf->leafbuflen[PETS… in PetscSFLinkStartCommunication_Persistent_Basic()
128 if (sf->leafbuflen[PETSCSF_REMOTE]) { in PetscSFLinkStartCommunication_MPIX_Stream()
/petsc/src/vec/is/sf/impls/basic/allgatherv/
H A Dsfallgatherv.c184 …->leafmtype_mpi]) PetscCall(PetscSFMalloc(sf, link->leafmtype_mpi, sf->leafbuflen[PETSCSF_REMOTE] … in PetscSFReduceBegin_Allgatherv()
238 …, leafdata, PETSC_MEMTYPE_HOST, link->leafbuf[PETSC_MEMTYPE_HOST], sf->leafbuflen[PETSCSF_REMOTE] … in PetscSFBcastToZero_Allgatherv()
/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu96 stmp[1] = sf->leafbuflen[PETSCSF_REMOTE]; in PetscSFSetUp_Basic_NVSHMEM()
218 …direction == PETSCSF_ROOT2LEAF) ? bas->rootbuflen[PETSCSF_REMOTE] : sf->leafbuflen[PETSCSF_REMOTE]; in PetscSFLinkBuildDependenceBegin()
233 …PetscInt buflen = (direction == PETSCSF_ROOT2LEAF) ? sf->leafbuflen[PETSCSF_REMOTE] : bas->r… in PetscSFLinkBuildDependenceEnd()
/petsc/include/petsc/private/
H A Dsfimpl.h83 …PetscInt leafbuflen[2]; /* Length (in unit) of leaf buffers, in layout of [PETSCSF_LOCAL/… member