Home
last modified time | relevance | path

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

/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx293 static PetscErrorCode ScatterAndOp(PetscSFLink link, PetscInt count, PetscInt srcStart, PetscSFPack… in ScatterAndOp() function
373 …PetscCall(ScatterAndOp<Type, Insert<Type>, BS, EQ>(link, count, srcStart, srcOpt, srcIdx, src, dst… in ScatterAndInsert()
415 link->d_ScatterAndAdd = ScatterAndOp<Type, Add<Type>, BS, EQ>; in PackInit_RealType()
416 link->d_ScatterAndMult = ScatterAndOp<Type, Mult<Type>, BS, EQ>; in PackInit_RealType()
417 link->d_ScatterAndMin = ScatterAndOp<Type, Min<Type>, BS, EQ>; in PackInit_RealType()
418 link->d_ScatterAndMax = ScatterAndOp<Type, Max<Type>, BS, EQ>; in PackInit_RealType()
428 link->da_ScatterAndInsert = ScatterAndOp<Type, AtomicInsert<Type>, BS, EQ>; in PackInit_RealType()
429 link->da_ScatterAndAdd = ScatterAndOp<Type, AtomicAdd<Type>, BS, EQ>; in PackInit_RealType()
430 link->da_ScatterAndMult = ScatterAndOp<Type, AtomicMult<Type>, BS, EQ>; in PackInit_RealType()
431 link->da_ScatterAndMin = ScatterAndOp<Type, AtomicMin<Type>, BS, EQ>; in PackInit_RealType()
[all …]
/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm_impl.hpp320 inline PetscErrorCode SfInterface<T>::ScatterAndOp(PetscSFLink link, PetscInt count, PetscInt srcSt… in ScatterAndOp() function in Petsc::sf::cupm::impl::SfInterface
370 …PetscCall(ScatterAndOp<Type, kernels::Insert<Type>, BS, EQ>(link, count, srcStart, srcOpt, srcIdx,… in ScatterAndInsert()
406 link->d_ScatterAndAdd = ScatterAndOp<Type, kernels::Add<Type>, BS, EQ>; in PackInit_RealType()
407 link->d_ScatterAndMult = ScatterAndOp<Type, kernels::Mult<Type>, BS, EQ>; in PackInit_RealType()
408 link->d_ScatterAndMin = ScatterAndOp<Type, kernels::Min<Type>, BS, EQ>; in PackInit_RealType()
409 link->d_ScatterAndMax = ScatterAndOp<Type, kernels::Max<Type>, BS, EQ>; in PackInit_RealType()
420 link->da_ScatterAndInsert = ScatterAndOp<Type, AtomicInsert<Type>, BS, EQ>; in PackInit_RealType()
421 link->da_ScatterAndAdd = ScatterAndOp<Type, AtomicAdd<Type>, BS, EQ>; in PackInit_RealType()
422 link->da_ScatterAndMult = ScatterAndOp<Type, AtomicMult<Type>, BS, EQ>; in PackInit_RealType()
423 link->da_ScatterAndMin = ScatterAndOp<Type, AtomicMin<Type>, BS, EQ>; in PackInit_RealType()
[all …]
H A Dsfcupm.hpp34 …static PetscErrorCode ScatterAndOp(PetscSFLink, PetscInt, PetscInt, PetscSFPackOpt, const PetscInt…
/petsc/src/vec/is/sf/impls/basic/
H A Dsfpack.c740 …k, PetscMemType mtype, MPI_Op op, PetscBool atomic, PetscErrorCode (**ScatterAndOp)(PetscSFLink, P… in PetscSFLinkGetScatterAndOp()
743 *ScatterAndOp = NULL; in PetscSFLinkGetScatterAndOp()
745 if (op == MPI_REPLACE) *ScatterAndOp = link->h_ScatterAndInsert; in PetscSFLinkGetScatterAndOp()
746 else if (op == MPI_SUM || op == MPIU_SUM) *ScatterAndOp = link->h_ScatterAndAdd; in PetscSFLinkGetScatterAndOp()
747 else if (op == MPI_PROD) *ScatterAndOp = link->h_ScatterAndMult; in PetscSFLinkGetScatterAndOp()
748 else if (op == MPI_MAX || op == MPIU_MAX) *ScatterAndOp = link->h_ScatterAndMax; in PetscSFLinkGetScatterAndOp()
749 else if (op == MPI_MIN || op == MPIU_MIN) *ScatterAndOp = link->h_ScatterAndMin; in PetscSFLinkGetScatterAndOp()
750 else if (op == MPI_LAND) *ScatterAndOp = link->h_ScatterAndLAND; in PetscSFLinkGetScatterAndOp()
751 else if (op == MPI_BAND) *ScatterAndOp = link->h_ScatterAndBAND; in PetscSFLinkGetScatterAndOp()
752 else if (op == MPI_LOR) *ScatterAndOp = link->h_ScatterAndLOR; in PetscSFLinkGetScatterAndOp()
[all …]
H A Dsfpack.h232 …AndOp(PetscSFLink, PetscMemType, MPI_Op, PetscBool, PetscErrorCode (**ScatterAndOp)(PetscSFLink, P…