This repository has been archived by the owner on Apr 15, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
Kokkos::Sum
Christian Trott edited this page May 1, 2020
·
1 revision
Specific implementation of ReducerConcept performing an add
operation
Header File: Kokkos_Core.hpp
Usage:
T result;
parallel_reduce(N,Functor,Sum<T,S>(result));
.
template<class Scalar, class Space>
class Sum{
public:
typedef Sum reducer;
typedef typename std::remove_cv<Scalar>::type value_type;
typedef Kokkos::View<value_type, Space> result_view_type;
KOKKOS_INLINE_FUNCTION
void join(value_type& dest, const value_type& src) const
KOKKOS_INLINE_FUNCTION
void join(volatile value_type& dest, const volatile value_type& src) const;
KOKKOS_INLINE_FUNCTION
void init( value_type& val) const;
KOKKOS_INLINE_FUNCTION
value_type& reference() const;
KOKKOS_INLINE_FUNCTION
result_view_type view() const;
KOKKOS_INLINE_FUNCTION
Sum(value_type& value_);
KOKKOS_INLINE_FUNCTION
Sum(const result_view_type& value_);
};
-
reducer
: The self type. -
value_type
: The reduction scalar type. -
result_view_type
: AKokkos::View
referencing the reduction result
-
Sum(value_type& result)
Constructs a reducer which references a local variable as its result location.
-
Sum(const result_view_type result)`
Constructs a reducer which references a specific view as its result location.
-
void join(value_type& dest, const value_type& src) const;`
Add
src
intodest
:dest+=src;
. -
void join(volatile value_type& dest, const volatile value_type& src) const;
Add
src
intodest
:dest+=src;
. -
void init( value_type& val) const;
Initialize
val
using the Kokkos::reduction_identity::sum() method. The default implementation setsval=0
. -
value_type& reference() const;
Returns a reference to the result provided in class constructor.
-
result_view_type view() const;
Returns a view of the result place provided in class constructor.
-
Sum<T,S>::value_type
is non-constT
-
Sum<T,S>::result_view_type
isKokkos::View<T,S,Kokkos::MemoryTraits<Kokkos::Unmanaged>>
. Note that the S (memory space) must be the same as the space where the result resides. - Requires:
Scalar
hasoperator =
andoperator +=
defined.Kokkos::reduction_identity<Scalar>::sum()
is a valid expression. - In order to use Sum with a custom type, a template specialization of Kokkos::reduction_identity must be defined. See Using Built-in Reducers with Custom Scalar Types for details
Home:
- Introduction
- Machine Model
- Programming Model
- Compiling
- Initialization
- View
- Parallel Dispatch
- Hierarchical Parallelism
- Custom Reductions
- Atomic Operations
- Subviews
- Interoperability
- Kokkos and Virtual Functions
- Initialization and Finalization
- View
- Data Parallelism
- Execution Policies
- Spaces
- Task Parallelism
- Utilities
- STL Compatibility
- Numerics
- Detection Idiom