DeviceReduce< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, PropagateNan, OutputIndex > Struct Template Reference

• ck
• tensor_operation
• device
• DeviceReduce

Public Member Functions | Static Public Attributes | List of all members

ck::tensor_operation::device::DeviceReduce< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, PropagateNan, OutputIndex > Struct Template Referenceabstract

#include <device_reduce.hpp>

Inheritance diagram for ck::tensor_operation::device::DeviceReduce< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, PropagateNan, OutputIndex >:

Public Member Functions
virtual std::unique_ptr< BaseArgument >	MakeArgumentPointer (const std::array< index_t, Rank > inLengths, const std::array< index_t, Rank > inStrides, const std::array< index_t, NumOutDim > outLengths, const std::array< index_t, NumOutDim > outStrides, const std::array< int, NumReduceDim > reduceDims, double alpha, double beta, const void *in_dev, const void *in_index_dev, void *out_dev, void *out_index_dev, const InElementwiseOperation in_elementwise_op, const AccElementwiseOperation acc_elementwise_op)=0
virtual std::unique_ptr< BaseInvoker >	MakeInvokerPointer ()=0
Public Member Functions inherited from ck::tensor_operation::device::BaseOperator
	BaseOperator ()=default
	BaseOperator (const BaseOperator &)=default
BaseOperator &	operator= (const BaseOperator &)=default
virtual bool	IsSupportedArgument (const BaseArgument *)
virtual std::string	GetTypeString () const
virtual std::string	GetInstanceString () const
virtual std::string	GetTypeIdName () const
virtual std::optional< std::string >	GetObjectName () const
virtual std::optional< std::string >	GetTemplateInfo () const
virtual std::string	GetTypeIdHashCode () const
virtual size_t	GetWorkSpaceSize (const BaseArgument *) const
virtual void	SetWorkSpacePointer (BaseArgument *p_arg, void *p_workspace, const StreamConfig &=StreamConfig{}) const
virtual	~BaseOperator ()

Static Public Attributes
static constexpr index_t	NumOutDim = (Rank - NumReduceDim == 0) ? 1 : Rank - NumReduceDim

Member Function Documentation

◆ MakeArgumentPointer()

template<typename InDataType, typename AccDataType, typename OutDataType, index_t Rank, index_t NumReduceDim, typename ReduceOperation, typename InElementwiseOperation, typename AccElementwiseOperation, bool PropagateNan, bool OutputIndex>

virtual std::unique_ptr< BaseArgument > ck::tensor_operation::device::DeviceReduce< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, PropagateNan, OutputIndex >::MakeArgumentPointer ( const std::array< index_t, Rank > inLengths, const std::array< index_t, Rank > inStrides, const std::array< index_t, NumOutDim > outLengths, const std::array< index_t, NumOutDim > outStrides, const std::array< int, NumReduceDim > reduceDims, double alpha, double beta, const void * in_dev, const void * in_index_dev, void * out_dev, void * out_index_dev, const InElementwiseOperation in_elementwise_op, const AccElementwiseOperation acc_elementwise_op )

pure virtual

◆ MakeInvokerPointer()

template<typename InDataType, typename AccDataType, typename OutDataType, index_t Rank, index_t NumReduceDim, typename ReduceOperation, typename InElementwiseOperation, typename AccElementwiseOperation, bool PropagateNan, bool OutputIndex>

virtual std::unique_ptr< BaseInvoker > ck::tensor_operation::device::DeviceReduce< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, PropagateNan, OutputIndex >::MakeInvokerPointer ( )

pure virtual

Implemented in ck::tensor_operation::device::DeviceReduceMultiBlock< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, OutMemoryDataOperation, PropagateNan, OutputIndex, HaveIndexInputIfOutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcVectorDim, InSrcVectorSize, OutDstVectorSize >, and ck::tensor_operation::device::DeviceReduceThreadWise< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, PropagateNan, OutputIndex, TransformIndexKtoGlobal, HaveIndexInputIfOutputIndex, BlockSize, MThreadSliceSize, KThreadSliceSize, InSrcVectorDim, InSrcVectorSize, OutDstVectorSize >.

Member Data Documentation

◆ NumOutDim

template<typename InDataType, typename AccDataType, typename OutDataType, index_t Rank, index_t NumReduceDim, typename ReduceOperation, typename InElementwiseOperation, typename AccElementwiseOperation, bool PropagateNan, bool OutputIndex>

index_t ck::tensor_operation::device::DeviceReduce< InDataType, AccDataType, OutDataType, Rank, NumReduceDim, ReduceOperation, InElementwiseOperation, AccElementwiseOperation, PropagateNan, OutputIndex >::NumOutDim = (Rank - NumReduceDim == 0) ? 1 : Rank - NumReduceDim

staticconstexpr

---

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

• device_reduce.hpp