Invoker Struct Reference

Invoker Struct Reference#

Composable Kernel: ck::tensor_operation::device::DeviceGroupedConvBwdWeight_Explicit_Xdl< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation, DeviceGemmV3Op >::Invoker Struct Reference

#include <device_grouped_conv_bwd_weight_explicit_xdl.hpp>

Inheritance diagram for ck::tensor_operation::device::DeviceGroupedConvBwdWeight_Explicit_Xdl< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation, DeviceGemmV3Op >::Invoker:

Public Types
using	Argument = DeviceOp::Argument
using	GemmArgument = typename DeviceGemmV3Op::Argument

Public Member Functions
float	Run (const Argument &arg, const StreamConfig &stream_config=StreamConfig{})
float	Run (const BaseArgument *p_arg, const StreamConfig &stream_config=StreamConfig{}) override
Public Member Functions inherited from ck::tensor_operation::device::BaseInvoker
	BaseInvoker ()=default
	BaseInvoker (const BaseInvoker &)=default
BaseInvoker &	operator= (const BaseInvoker &)=default
virtual	~BaseInvoker ()

Public Attributes
DeviceGemmV3Op::Invoker	explicit_gemm_op

Member Typedef Documentation

◆ Argument

template<ck::index_t NDimSpatial, typename InLayout, typename WeiLayout, typename OutLayout, typename InDataType, typename WeiDataType, typename OutDataType, typename InElementwiseOperation, typename WeiElementwiseOperation, typename OutElementwiseOperation, typename DeviceGemmV3Op>

using ck::tensor_operation::device::DeviceGroupedConvBwdWeight_Explicit_Xdl< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation, DeviceGemmV3Op >::Invoker::Argument = DeviceOp::Argument

◆ GemmArgument

template<ck::index_t NDimSpatial, typename InLayout, typename WeiLayout, typename OutLayout, typename InDataType, typename WeiDataType, typename OutDataType, typename InElementwiseOperation, typename WeiElementwiseOperation, typename OutElementwiseOperation, typename DeviceGemmV3Op>

using ck::tensor_operation::device::DeviceGroupedConvBwdWeight_Explicit_Xdl< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation, DeviceGemmV3Op >::Invoker::GemmArgument = typename DeviceGemmV3Op::Argument

Member Function Documentation

◆ Run() [1/2]

template<ck::index_t NDimSpatial, typename InLayout, typename WeiLayout, typename OutLayout, typename InDataType, typename WeiDataType, typename OutDataType, typename InElementwiseOperation, typename WeiElementwiseOperation, typename OutElementwiseOperation, typename DeviceGemmV3Op>

float ck::tensor_operation::device::DeviceGroupedConvBwdWeight_Explicit_Xdl< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation, DeviceGemmV3Op >::Invoker::Run	(	const Argument &	arg,
		const StreamConfig &	stream_config = StreamConfig{} )

inline

◆ Run() [2/2]

template<ck::index_t NDimSpatial, typename InLayout, typename WeiLayout, typename OutLayout, typename InDataType, typename WeiDataType, typename OutDataType, typename InElementwiseOperation, typename WeiElementwiseOperation, typename OutElementwiseOperation, typename DeviceGemmV3Op>

float ck::tensor_operation::device::DeviceGroupedConvBwdWeight_Explicit_Xdl< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation, DeviceGemmV3Op >::Invoker::Run	(	const BaseArgument *	p_arg,
		const StreamConfig &	stream_config = StreamConfig{} )

inlineoverridevirtual

Reimplemented from ck::tensor_operation::device::BaseInvoker.

Member Data Documentation

◆ explicit_gemm_op

template<ck::index_t NDimSpatial, typename InLayout, typename WeiLayout, typename OutLayout, typename InDataType, typename WeiDataType, typename OutDataType, typename InElementwiseOperation, typename WeiElementwiseOperation, typename OutElementwiseOperation, typename DeviceGemmV3Op>

DeviceGemmV3Op::Invoker ck::tensor_operation::device::DeviceGroupedConvBwdWeight_Explicit_Xdl< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation, DeviceGemmV3Op >::Invoker::explicit_gemm_op

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

device_grouped_conv_bwd_weight_explicit_xdl.hpp