Home
last modified time | relevance | path

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

/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm.hpp31 …static PetscErrorCode FetchAndOp(PetscSFLink, PetscInt, PetscInt, PetscSFPackOpt, const PetscInt *…
H A Dsfcupm_impl.hpp308 inline PetscErrorCode SfInterface<T>::FetchAndOp(PetscSFLink link, PetscInt count, PetscInt start, … in FetchAndOp() function in Petsc::sf::cupm::impl::SfInterface
402 link->d_FetchAndAdd = FetchAndOp<Type, kernels::Add<Type>, BS, EQ>; in PackInit_RealType()
418 link->da_FetchAndAdd = FetchAndOp<Type, AtomicAdd<Type>, BS, EQ>; in PackInit_RealType()
445 link->da_FetchAndAdd = FetchAndOp<Type, AtomicAdd<Type>, BS, EQ>; in Init()
485 link->d_FetchAndAdd = FetchAndOp<Type, kernels::Add<Type>, BS, EQ>; in PackInit_IntegerType()
511 link->d_FetchAndAdd = FetchAndOp<Type, kernels::Add<Type>, BS, EQ>; in PackInit_ComplexType()
/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx274 static PetscErrorCode FetchAndOp(PetscSFLink link, PetscInt count, PetscInt start, PetscSFPackOpt o… in FetchAndOp() function
412 link->d_FetchAndAdd = FetchAndOp<Type, Add<Type>, BS, EQ>; in PackInit_RealType()
426 link->da_FetchAndAdd = FetchAndOp<Type, AtomicFetchAdd<Type>, BS, EQ>; in PackInit_RealType()
451 link->d_FetchAndAdd = FetchAndOp<Type, Add<Type>, BS, EQ>; in PackInit_IntegerType()
476 link->da_FetchAndAdd = FetchAndOp<Type, AtomicFetchAdd<Type>, BS, EQ>; in PackInit_IntegerType()
499 link->d_FetchAndAdd = FetchAndOp<Type, Add<Type>, BS, EQ>; in PackInit_ComplexType()
509 link->da_FetchAndAdd = FetchAndOp<Type, AtomicFetchAdd<Type>, BS, EQ>; in PackInit_ComplexType()
/petsc/src/vec/is/sf/impls/basic/
H A Dsfpack.c793 …nk, PetscMemType mtype, MPI_Op op, PetscBool atomic, PetscErrorCode (**FetchAndOp)(PetscSFLink, Pe… in PetscSFLinkGetFetchAndOp()
796 *FetchAndOp = NULL; in PetscSFLinkGetFetchAndOp()
798 if (PetscMemTypeHost(mtype)) *FetchAndOp = link->h_FetchAndAdd; in PetscSFLinkGetFetchAndOp()
800 else if (PetscMemTypeDevice(mtype) && !atomic) *FetchAndOp = link->d_FetchAndAdd; in PetscSFLinkGetFetchAndOp()
801 else if (PetscMemTypeDevice(mtype) && atomic) *FetchAndOp = link->da_FetchAndAdd; in PetscSFLinkGetFetchAndOp()
1065 …PetscErrorCode (*FetchAndOp)(PetscSFLink, PetscInt, PetscInt, PetscSFPackOpt, const PetscInt *, vo… in PetscSFLinkFetchAndOpRemote() local
1073 …tscCall(PetscSFLinkGetFetchAndOp(link, rootmtype, op, bas->rootdups[PETSCSF_REMOTE], &FetchAndOp)); in PetscSFLinkFetchAndOpRemote()
1075 …PetscCall((*FetchAndOp)(link, count, start, opt, rootindices, rootdata, link->rootbuf[PETSCSF_REMO… in PetscSFLinkFetchAndOpRemote()
H A Dsfpack.h231 …hAndOp(PetscSFLink, PetscMemType, MPI_Op, PetscBool, PetscErrorCode (**FetchAndOp)(PetscSFLink, Pe…