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:

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