DeviceGemmMultipleDSplitKWrapper< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CDEElementwiseOperation > Struct Template Reference

DeviceGemmMultipleDSplitKWrapper< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CDEElementwiseOperation > Struct Template Reference#

Composable Kernel: ck::tensor_operation::device::DeviceGemmMultipleDSplitKWrapper< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CDEElementwiseOperation > Struct Template Reference

Wrapper for backward compatibility that allows to use instances of DeviceGemmMultipleDSplitK in contexts where DeviceGemmMultipleD is expected. More...

#include <device_gemm_multiple_d.hpp>

Inheritance diagram for ck::tensor_operation::device::DeviceGemmMultipleDSplitKWrapper< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CDEElementwiseOperation >:

Public Types
using	DeviceOp

Public Member Functions
	DeviceGemmMultipleDSplitKWrapper (std::unique_ptr< DeviceOp > p_op)
bool	IsSupportedArgument (const BaseArgument *p_arg) override
std::unique_ptr< BaseArgument >	MakeArgumentPointer (const void p_a, const void p_b, std::array< const void , NumDTensor > p_ds, void p_e, ck::index_t M, ck::index_t N, ck::index_t K, ck::index_t StrideA, ck::index_t StrideB, std::array< ck::index_t, NumDTensor > StrideDs, ck::index_t StrideE, AElementwiseOperation a_element_op, BElementwiseOperation b_element_op, CDEElementwiseOperation cde_element_op) override
std::unique_ptr< BaseInvoker >	MakeInvokerPointer () override
std::string	GetTypeString () const 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	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 ()

Static Public Attributes
static constexpr index_t	NumDTensor = DsDataType::Size()
Static Public Attributes inherited from ck::tensor_operation::device::DeviceGemmMultipleD< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CDEElementwiseOperation >
static constexpr index_t	NumDTensor = DsDataType::Size()

Detailed Description

template<typename ALayout, typename BLayout, typename DsLayout, typename ELayout, typename ADataType, typename BDataType, typename DsDataType, typename EDataType, typename AElementwiseOperation, typename BElementwiseOperation, typename CDEElementwiseOperation>
struct ck::tensor_operation::device::DeviceGemmMultipleDSplitKWrapper< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CDEElementwiseOperation >

Wrapper for backward compatibility that allows to use instances of DeviceGemmMultipleDSplitK in contexts where DeviceGemmMultipleD is expected.

Note: The main area where it can be used is DeviceOperationInstanceFactory::GetInstances(). The only difference between API of DeviceGemmMultipleD and DeviceGemmMultipleDSplitK is that DeviceGemmMultipleDSplitK::MakeArgumentPointer requires an additional parameter KBatch which is explicitly passed as 1 by this wrapper.