Skip to content

Kokkos::Prod

Christian Trott edited this page May 1, 2020 · 1 revision

Kokkos::Prod

Specific implementation of ReducerConcept performing a multiply operation

Header File: Kokkos_Core.hpp

Usage:

T result;
parallel_reduce(N,Functor,Prod<T,S>(result));

.

Synopsis

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_);
};

Public Class Members

Typedefs

  • reducer: The self type.
  • value_type: The reduction scalar type.
  • result_view_type: A Kokkos::View referencing the reduction result

Constructors

  • 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.

Functions

  • void join(value_type& dest, const value_type& src)  const;`

    Multiply src and dest and store in dest: dest*=src;.

  • void join(volatile value_type& dest, const volatile value_type& src) const;

    Multiply src and dest and store in dest: dest*=src;.

  • void init( value_type& val)  const;

    Initialize val using the Kokkos::reduction_identity::prod() method. The default implementation sets val=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.

Additional Information

  • Prod<T,S>::value_type is non-const T
  • Prod<T,S>::result_view_type is Kokkos::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 has operator = and operator *= 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
Clone this wiki locally