DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize > Struct Template Reference

DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize > Struct Template Reference#

Composable Kernel: ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize > Struct Template Reference

#include <device_put_element_impl.hpp>

Inheritance diagram for ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >:

Classes
struct	Argument

struct	Invoker

Public Types
using	InGrid1dDesc = decltype(MakeDescriptor_M(1, 1, 1))

using	GridwisePutElement = GridwisePutElement_1D< InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >

Public Member Functions
bool	IsSupportedArgument (const BaseArgument *p_arg) override

std::unique_ptr< BaseArgument >	MakeArgumentPointer (const void p_input, const void p_indices, void *p_output, index_t input_length, index_t, ElementwiseOperation elementwise_op) override

std::unique_ptr< BaseInvoker >	MakeInvokerPointer () override

Public Member Functions inherited from ck::tensor_operation::device::BaseOperator
	BaseOperator ()=default

	BaseOperator (const BaseOperator &)=default

BaseOperator &	operator= (const BaseOperator &)=default

virtual std::string	GetTypeString () 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 Member Functions
template<typename Desc_M >
static auto	PadDescriptor_M_1d (Desc_M desc_m, index_t gridSize, index_t blockSize)

static auto	MakeDescriptor_M (index_t length, index_t gridSize, index_t blockSize)

Member Typedef Documentation

◆ GridwisePutElement

template<typename InDataType , typename IndexDataType , typename OutDataType , typename ElementwiseOperation , InMemoryDataOperationEnum MemOp, ck::index_t InVectorSize>

using ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::GridwisePutElement = GridwisePutElement_1D<InGrid1dDesc, InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize>

◆ InGrid1dDesc

template<typename InDataType , typename IndexDataType , typename OutDataType , typename ElementwiseOperation , InMemoryDataOperationEnum MemOp, ck::index_t InVectorSize>

using ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::InGrid1dDesc = decltype(MakeDescriptor_M(1, 1, 1))

Member Function Documentation

◆ IsSupportedArgument()

template<typename InDataType , typename IndexDataType , typename OutDataType , typename ElementwiseOperation , InMemoryDataOperationEnum MemOp, ck::index_t InVectorSize>

bool ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::IsSupportedArgument ( const BaseArgument * p_arg )

inlineoverridevirtual

Reimplemented from ck::tensor_operation::device::BaseOperator.

◆ MakeArgumentPointer()

template<typename InDataType , typename IndexDataType , typename OutDataType , typename ElementwiseOperation , InMemoryDataOperationEnum MemOp, ck::index_t InVectorSize>

std::unique_ptr<BaseArgument> ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::MakeArgumentPointer	(	const void *	p_input,
		const void *	p_indices,
		void *	p_output,
		index_t	input_length,
		index_t	,
		ElementwiseOperation	elementwise_op
	)

inlineoverridevirtual

Implements ck::tensor_operation::device::DevicePutElement< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp >.

◆ MakeDescriptor_M()

template<typename InDataType , typename IndexDataType , typename OutDataType , typename ElementwiseOperation , InMemoryDataOperationEnum MemOp, ck::index_t InVectorSize>

static auto ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::MakeDescriptor_M	(	index_t	length,
		index_t	gridSize,
		index_t	blockSize
	)

inlinestatic

◆ MakeInvokerPointer()

template<typename InDataType , typename IndexDataType , typename OutDataType , typename ElementwiseOperation , InMemoryDataOperationEnum MemOp, ck::index_t InVectorSize>

std::unique_ptr<BaseInvoker> ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::MakeInvokerPointer ( )

inlineoverridevirtual

Implements ck::tensor_operation::device::DevicePutElement< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp >.

◆ PadDescriptor_M_1d()

template<typename InDataType , typename IndexDataType , typename OutDataType , typename ElementwiseOperation , InMemoryDataOperationEnum MemOp, ck::index_t InVectorSize>

template<typename Desc_M >

static auto ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::PadDescriptor_M_1d	(	Desc_M	desc_m,
		index_t	gridSize,
		index_t	blockSize
	)

inlinestatic

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

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-7.0.1/include/ck/tensor_operation/gpu/device/impl/device_put_element_impl.hpp