Argument Struct Reference

Argument Struct Reference#

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

#include <device_put_element_impl.hpp>

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

Public Member Functions
	Argument (const InDataType p_input, const IndexDataType p_indices, OutDataType *p_output, index_t input_length, ElementwiseOperation elementwise_op)
Public Member Functions inherited from ck::tensor_operation::device::BaseArgument
	BaseArgument ()=default
	BaseArgument (const BaseArgument &)=default
BaseArgument &	operator= (const BaseArgument &)=default
virtual	~BaseArgument ()

Public Attributes
const InDataType *	p_input_
const IndexDataType *	p_indices_
OutDataType *	p_output_
index_t	input_length_raw_
ElementwiseOperation	elementwise_op_
index_t	blockSize_
Public Attributes inherited from ck::tensor_operation::device::BaseArgument
void *	p_workspace_ = nullptr

Constructor & Destructor Documentation

◆ Argument()

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

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

inline

Member Data Documentation

◆ blockSize_

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

index_t ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::Argument::blockSize_

◆ elementwise_op_

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

ElementwiseOperation ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::Argument::elementwise_op_

◆ input_length_raw_

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

index_t ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::Argument::input_length_raw_

◆ p_indices_

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

const IndexDataType* ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::Argument::p_indices_

◆ p_input_

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

const InDataType* ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::Argument::p_input_

◆ p_output_

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

OutDataType* ck::tensor_operation::device::DevicePutElementImpl< InDataType, IndexDataType, OutDataType, ElementwiseOperation, MemOp, InVectorSize >::Argument::p_output_

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

device_put_element_impl.hpp