Home
last modified time | relevance | path

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

/petsc/src/vec/is/sf/impls/basic/allgather/
H A Dsfallgather.c75 …leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi]) PetscCall(PetscSFMalloc(sf, link->leafmtype_mp… in PetscSFReduceBegin_Allgather()
76 …if (rank == 0 && link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi] == leafbuf) leafbuf = MP… in PetscSFReduceBegin_Allgather()
79 …PetscCallMPI(MPI_Reduce(leafbuf, link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], count, … in PetscSFReduceBegin_Allgather()
80 …PetscCallMPI(MPIU_Iscatter(link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], recvcount, un… in PetscSFReduceBegin_Allgather()
/petsc/src/vec/is/sf/impls/basic/
H A Dsfmpi.c201 … (!link->leafbuf_alloc[i][leafmtype]) PetscCall(PetscSFMalloc(sf, leafmtype, sf->leafbuflen[i] * l… in PetscSFLinkCreate_MPI()
202 link->leafbuf[i][leafmtype] = link->leafbuf_alloc[i][leafmtype]; in PetscSFLinkCreate_MPI()
214 …ink->leafbuf_alloc[PETSCSF_REMOTE][PETSC_MEMTYPE_HOST]) PetscCall(PetscMalloc(sf->leafbuflen[PETSC… in PetscSFLinkCreate_MPI()
215 …link->leafbuf[PETSCSF_REMOTE][PETSC_MEMTYPE_HOST] = link->leafbuf_alloc[PETSCSF_REMOTE][PETSC_MEMT… in PetscSFLinkCreate_MPI()
H A Dsfpack.h190 char *leafbuf_alloc[2][2]; member
H A Dsfpack.c413 PetscCall(PetscFree(link->leafbuf_alloc[i][PETSC_MEMTYPE_HOST])); in PetscSFLinkDestroy()
/petsc/src/vec/is/sf/impls/basic/allgatherv/
H A Dsfallgatherv.c184 …leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi]) PetscCall(PetscSFMalloc(sf, link->leafmtype_mp… in PetscSFReduceBegin_Allgatherv()
186 …if (rank == 0 && link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi] == leafbuf) leafbuf = MP… in PetscSFReduceBegin_Allgatherv()
188 …PetscCallMPI(MPI_Reduce(leafbuf, link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], count, … in PetscSFReduceBegin_Allgatherv()
189 …PetscCallMPI(MPIU_Iscatterv(link->leafbuf_alloc[PETSCSF_REMOTE][link->leafmtype_mpi], dat->recvcou… in PetscSFReduceBegin_Allgatherv()
/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu678 PetscCall(PetscNvshmemFree(link->leafbuf_alloc[PETSCSF_REMOTE][PETSC_MEMTYPE_DEVICE])); in PetscSFLinkDestroy_NVSHMEM()
778 …>leafbuf_alloc[PETSCSF_REMOTE][PETSC_MEMTYPE_DEVICE]) PetscCall(PetscNvshmemMalloc(sf->leafbuflen_… in PetscSFLinkCreate_NVSHMEM()
779 …link->leafbuf[PETSCSF_REMOTE][PETSC_MEMTYPE_DEVICE] = link->leafbuf_alloc[PETSCSF_REMOTE][PETSC_ME… in PetscSFLinkCreate_NVSHMEM()
/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx632 PetscCall(PetscSFFree(sf, PETSC_MEMTYPE_DEVICE, link->leafbuf_alloc[i][PETSC_MEMTYPE_DEVICE])); in PetscSFLinkDestroy_Kokkos()
/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm_impl.hpp631 PetscCallCUPM(cupmFree(link->leafbuf_alloc[i][PETSC_MEMTYPE_DEVICE])); in LinkDestroy_MPI()