Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType > Struct Template Reference

#include <TensorReductionSycl.h>

Public Types
typedef OpDefiner< OpType, OutScalar, Index, false >	OpDef
typedef OpDef::type	Op
typedef cl::sycl::accessor< OutScalar, 1, cl::sycl::access::mode::read_write, cl::sycl::access::target::local >	ScratchAccessor

Public Member Functions
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE	SecondStepPartialReduction (ScratchAccessor, InputAccessor input_accessor_, OutputAccessor output_accessor_, OpType op_, const Index num_coeffs_to_preserve_, const Index num_coeffs_to_reduce_)
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void	operator() (cl::sycl::nd_item< 1 > itemID) const

Public Attributes
InputAccessor	input_accessor
OutputAccessor	output_accessor
Op	op
const Index	num_coeffs_to_preserve
const Index	num_coeffs_to_reduce

Member Typedef Documentation

Op

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

typedef OpDef::type Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::Op

OpDef

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

typedef OpDefiner<OpType, OutScalar, Index, false> Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::OpDef

ScratchAccessor

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

typedef cl::sycl::accessor<OutScalar, 1, cl::sycl::access::mode::read_write, cl::sycl::access::target::local> Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::ScratchAccessor

Constructor & Destructor Documentation

SecondStepPartialReduction()

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::SecondStepPartialReduction ( ScratchAccessor  , InputAccessor  input_accessor_, OutputAccessor  output_accessor_, OpType  op_, const Index  num_coeffs_to_preserve_, const Index  num_coeffs_to_reduce_  )

inline

      : input_accessor(input_accessor_),
        output_accessor(output_accessor_),
        op(OpDef::get_op(op_)),
        num_coeffs_to_preserve(num_coeffs_to_preserve_),
        num_coeffs_to_reduce(num_coeffs_to_reduce_) {}

Member Function Documentation

operator()()

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::operator() ( cl::sycl::nd_item< 1 >  itemID ) const

inline

                                                                                       {
    const Index globalId = itemID.get_global_id(0);
 
    if (globalId >= num_coeffs_to_preserve) return;
 
    auto in_ptr = input_accessor + globalId;
 
    OutScalar accumulator = op.initialize();
    // num_coeffs_to_reduce is not bigger that 256
    for (Index i = 0; i < num_coeffs_to_reduce; i++) {
      op.reduce(*in_ptr, &accumulator);
      in_ptr += num_coeffs_to_preserve;
    }
    output_accessor[globalId] = op.finalize(accumulator);
  }

References i, Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::input_accessor, Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::num_coeffs_to_preserve, Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::num_coeffs_to_reduce, Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::op, and Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::output_accessor.

Member Data Documentation

input_accessor

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

InputAccessor Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::input_accessor

Referenced by Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::operator()().

num_coeffs_to_preserve

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

const Index Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::num_coeffs_to_preserve

Referenced by Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::operator()().

num_coeffs_to_reduce

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

const Index Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::num_coeffs_to_reduce

Referenced by Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::operator()().

op

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

Op Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::op

Referenced by Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::operator()().

output_accessor

template<typename OutScalar , typename Index , typename InputAccessor , typename OutputAccessor , typename OpType >

OutputAccessor Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::output_accessor

Referenced by Eigen::TensorSycl::internal::SecondStepPartialReduction< OutScalar, Index, InputAccessor, OutputAccessor, OpType >::operator()().

---

The documentation for this struct was generated from the following file:

• TensorReductionSycl.h