forked from kokkos/kokkos
-
Notifications
You must be signed in to change notification settings - Fork 0
Kokkos::Prod
Christian Trott edited this page May 1, 2020
·
1 revision
Specific implementation of ReducerConcept performing a multiply
operation
Header File: Kokkos_Core.hpp
Usage:
T result;
parallel_reduce(N,Functor,Prod<T,S>(result));
.
template<class Scalar, class Space>
class Prod{
public:
typedef Prod 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
Prod(value_type& value_);
KOKKOS_INLINE_FUNCTION
Prod(const result_view_type& value_);
};
-
reducer
: The self type. -
value_type
: The reduction scalar type. -
result_view_type
: AKokkos::View
referencing the reduction result
-
Prod(value_type& result)
Constructs a reducer which references a local variable as its result location.
-
Prod(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;`
Multiply
src
anddest
and store indest
:dest*=src;
. -
void join(volatile value_type& dest, const volatile value_type& src) const;
Multiply
src
anddest
and store indest
:dest*=src;
. -
void init( value_type& val) const;
Initialize
val
using the Kokkos::reduction_identity::prod() method. The default implementation setsval=1
. -
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.
-
Prod<T,S>::value_type
is non-constT
-
Prod<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>::prod()
is a valid expression. - In order to use Prod 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