DevicePermute< NumDim, InDataType, OutDataType, ElementwiseOperation > Struct Template Reference

• ck
• tensor_operation
• device
• DevicePermute

Public Types | Public Member Functions | List of all members

ck::tensor_operation::device::DevicePermute< NumDim, InDataType, OutDataType, ElementwiseOperation > Struct Template Referenceabstract

#include <device_permute.hpp>

Inheritance diagram for ck::tensor_operation::device::DevicePermute< NumDim, InDataType, OutDataType, ElementwiseOperation >:

Public Types
using	Lengths = std::array<index_t, NumDim>
using	Strides = Lengths

Public Member Functions
virtual std::unique_ptr< BaseArgument >	MakeArgumentPointer (const Lengths &in_lengths, const Strides &in_strides, const Lengths &out_lengths, const Strides &out_strides, const void *in_dev_buffer, void *out_dev_buffer, ElementwiseOperation 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 ()

Member Typedef Documentation

◆ Lengths

template<index_t NumDim, typename InDataType, typename OutDataType, typename ElementwiseOperation>

using ck::tensor_operation::device::DevicePermute< NumDim, InDataType, OutDataType, ElementwiseOperation >::Lengths = std::array<index_t, NumDim>

◆ Strides

template<index_t NumDim, typename InDataType, typename OutDataType, typename ElementwiseOperation>

using ck::tensor_operation::device::DevicePermute< NumDim, InDataType, OutDataType, ElementwiseOperation >::Strides = Lengths

Member Function Documentation

◆ MakeArgumentPointer()

template<index_t NumDim, typename InDataType, typename OutDataType, typename ElementwiseOperation>

virtual std::unique_ptr< BaseArgument > ck::tensor_operation::device::DevicePermute< NumDim, InDataType, OutDataType, ElementwiseOperation >::MakeArgumentPointer ( const Lengths & in_lengths, const Strides & in_strides, const Lengths & out_lengths, const Strides & out_strides, const void * in_dev_buffer, void * out_dev_buffer, ElementwiseOperation elementwise_op )

pure virtual

Implemented in ck::tensor_operation::device::DevicePermuteImpl< NumDim, InDataType, OutDataType, ElementwiseOperation, BlockSize, NPerBlock, HPerBlock, WPerBlock, InBlockLdsExtraW, InBlockTransferThreadClusterLengths, InBlockTransferThreadClusterArrangeOrder, SrcVectorDim, DstVectorDim, SrcScalarPerVector, DstScalarPerVector >.

◆ MakeInvokerPointer()

template<index_t NumDim, typename InDataType, typename OutDataType, typename ElementwiseOperation>

virtual std::unique_ptr< BaseInvoker > ck::tensor_operation::device::DevicePermute< NumDim, InDataType, OutDataType, ElementwiseOperation >::MakeInvokerPointer ( )

pure virtual

Implemented in ck::tensor_operation::device::DevicePermuteImpl< NumDim, InDataType, OutDataType, ElementwiseOperation, BlockSize, NPerBlock, HPerBlock, WPerBlock, InBlockLdsExtraW, InBlockTransferThreadClusterLengths, InBlockTransferThreadClusterArrangeOrder, SrcVectorDim, DstVectorDim, SrcScalarPerVector, DstScalarPerVector >.

---

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

• device_permute.hpp