DeviceGroupedGemmFixedNK< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation > Struct Template Reference#
ck::tensor_operation::device::DeviceGroupedGemmFixedNK< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation > Struct Template Reference
#include <device_grouped_gemm_fixed_nk.hpp>
Inheritance diagram for ck::tensor_operation::device::DeviceGroupedGemmFixedNK< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation >:
Additional Inherited Members | |
| Public Member Functions inherited from ck::tensor_operation::device::DeviceGroupedGemmSplitK< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation > | |
| virtual void | SetKBatchSize (BaseArgument *p_arg, index_t kbatch) const =0 |
| Sets the k batch size. | |
| virtual void | SetKBatch (BaseArgument *p_arg, index_t kbatch) const |
| Sets the k batch size. | |
| Public Member Functions inherited from ck::tensor_operation::device::DeviceGroupedGemm< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation > | |
| virtual std::unique_ptr< BaseArgument > | MakeArgumentPointer (std::vector< const void * > &p_a, std::vector< const void * > &p_b, std::vector< std::array< const void *, NumDTensor > > &p_ds, std::vector< void * > &p_e, std::vector< GemmDesc > &gemm_desc, AElementwiseOperation a_element_op, BElementwiseOperation b_element_op, CElementwiseOperation c_element_op)=0 |
| virtual std::unique_ptr< BaseInvoker > | MakeInvokerPointer ()=0 |
| virtual void | SetDeviceKernelArgs (BaseArgument *p_arg, void *p_dev_kernel_args, const void *p_host_kernel_args) const |
| Sets the device kernel arguments pointer and may copy data to device. | |
| virtual void | SetDeviceKernelArgs (BaseArgument *p_arg, void *p_dev_kernel_args) const |
| Sets the device kernel arguments pointer and may copy data to device. | |
| virtual size_t | GetDeviceKernelArgSize (const BaseArgument *p_arg) const |
| Gets the device kernel argument size. | |
| Public Member Functions inherited from ck::tensor_operation::device::BaseOperator | |
| BaseOperator ()=default | |
| BaseOperator (const BaseOperator &)=default | |
| BaseOperator & | operator= (const BaseOperator &)=default |
| virtual bool | IsSupportedArgument (const BaseArgument *) |
| virtual std::string | GetTypeString () const |
| 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 inherited from ck::tensor_operation::device::DeviceGroupedGemm< ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation > | |
| static constexpr index_t | NumDTensor = DsDataType::Size() |
The documentation for this struct was generated from the following file: