Home
last modified time | relevance | path

Searched refs:FetchAndOpLocal (Results 1 – 5 of 5) sorted by relevance

/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm.hpp40 …static PetscErrorCode FetchAndOpLocal(PetscSFLink, PetscInt, PetscInt, PetscSFPackOpt, const Petsc…
H A Dsfcupm_impl.hpp377 inline PetscErrorCode SfInterface<T>::FetchAndOpLocal(PetscSFLink link, PetscInt count, PetscInt ro… in FetchAndOpLocal() function in Petsc::sf::cupm::impl::SfInterface
410 link->d_FetchAndAddLocal = FetchAndOpLocal<Type, kernels::Add<Type>, BS, EQ>; in PackInit_RealType()
425 link->da_FetchAndAddLocal = FetchAndOpLocal<Type, AtomicAdd<Type>, BS, EQ>; in PackInit_RealType()
458 link->da_FetchAndAddLocal = FetchAndOpLocal<Type, AtomicAdd<Type>, BS, EQ>; in Init()
498 link->d_FetchAndAddLocal = FetchAndOpLocal<Type, kernels::Add<Type>, BS, EQ>; in PackInit_IntegerType()
516 link->d_FetchAndAddLocal = FetchAndOpLocal<Type, kernels::Add<Type>, BS, EQ>; in PackInit_ComplexType()
/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx379 static PetscErrorCode FetchAndOpLocal(PetscSFLink link, PetscInt count, PetscInt rootstart, PetscSF… in FetchAndOpLocal() function
419 link->d_FetchAndAddLocal = FetchAndOpLocal<Type, Add<Type>, BS, EQ>; in PackInit_RealType()
433 link->da_FetchAndAddLocal = FetchAndOpLocal<Type, AtomicFetchAdd<Type>, BS, EQ>; in PackInit_RealType()
464 link->d_FetchAndAddLocal = FetchAndOpLocal<Type, Add<Type>, BS, EQ>; in PackInit_IntegerType()
488 link->da_FetchAndAddLocal = FetchAndOpLocal<Type, AtomicFetchAdd<Type>, BS, EQ>; in PackInit_IntegerType()
504 link->d_FetchAndAddLocal = FetchAndOpLocal<Type, Add<Type>, BS, EQ>; in PackInit_ComplexType()
514 link->da_FetchAndAddLocal = FetchAndOpLocal<Type, AtomicFetchAdd<Type>, BS, EQ>; in PackInit_ComplexType()
/petsc/src/vec/is/sf/impls/basic/
H A Dsfpack.c806 …, PetscMemType mtype, MPI_Op op, PetscBool atomic, PetscErrorCode (**FetchAndOpLocal)(PetscSFLink,… in PetscSFLinkGetFetchAndOpLocal()
809 *FetchAndOpLocal = NULL; in PetscSFLinkGetFetchAndOpLocal()
811 if (PetscMemTypeHost(mtype)) *FetchAndOpLocal = link->h_FetchAndAddLocal; in PetscSFLinkGetFetchAndOpLocal()
813 else if (PetscMemTypeDevice(mtype) && !atomic) *FetchAndOpLocal = link->d_FetchAndAddLocal; in PetscSFLinkGetFetchAndOpLocal()
814 else if (PetscMemTypeDevice(mtype) && atomic) *FetchAndOpLocal = link->da_FetchAndAddLocal; in PetscSFLinkGetFetchAndOpLocal()
1151 …PetscErrorCode (*FetchAndOpLocal)(PetscSFLink, PetscInt, PetscInt, PetscSFPackOpt, const PetscInt … in PetscSFLinkFetchAndOpLocal() local
1161 …etscSFLinkGetFetchAndOpLocal(link, rootmtype, op, bas->rootdups[PETSCSF_LOCAL], &FetchAndOpLocal)); in PetscSFLinkFetchAndOpLocal()
1162 …PetscCall((*FetchAndOpLocal)(link, count, rootstart, rootopt, rootindices, rootdata, leafstart, le… in PetscSFLinkFetchAndOpLocal()
H A Dsfpack.h233 …ocal(PetscSFLink, PetscMemType, MPI_Op, PetscBool, PetscErrorCode (**FetchAndOpLocal)(PetscSFLink,…