DeviceGemm_Wmma_CShuffleV3_Common< GridwiseGemm, AsDataType, BsDataType, DsDataType, EDataType, MPerBlock, NPerBlock, KPerBlock, BlockSize, AK1, BK1, GemmSpec, CDEShuffleBlockTransferScalarPerVectors, BlkGemmPipeSched, BlkGemmPipelineVer, ComputeTypeA, ComputeTypeB > Struct Template Reference

• ck
• tensor_operation
• device
• DeviceGemm_Wmma_CShuffleV3_Common

Classes | Public Types | Static Public Member Functions | List of all members

ck::tensor_operation::device::DeviceGemm_Wmma_CShuffleV3_Common< GridwiseGemm, AsDataType, BsDataType, DsDataType, EDataType, MPerBlock, NPerBlock, KPerBlock, BlockSize, AK1, BK1, GemmSpec, CDEShuffleBlockTransferScalarPerVectors, BlkGemmPipeSched, BlkGemmPipelineVer, ComputeTypeA, ComputeTypeB > Struct Template Reference

#include <device_gemm_wmma_cshuffle_v3_common.hpp>

Classes
struct	Invoker
	Helper structure responsible for kernel invocation. More...

Public Types
using	Argument = typename GridwiseGemm::Argument

Static Public Member Functions
static constexpr bool	IsValidCompilationParameter ()
static bool	IsSupportedArgument (const Argument &arg)

Member Typedef Documentation

◆ Argument

template<typename GridwiseGemm, typename AsDataType, typename BsDataType, typename DsDataType, typename EDataType, index_t MPerBlock, index_t NPerBlock, index_t KPerBlock, index_t BlockSize, index_t AK1, index_t BK1, GemmSpecialization GemmSpec, typename CDEShuffleBlockTransferScalarPerVectors, BlockGemmPipelineScheduler BlkGemmPipeSched, BlockGemmPipelineVersion BlkGemmPipelineVer, typename ComputeTypeA, typename ComputeTypeB>

using ck::tensor_operation::device::DeviceGemm_Wmma_CShuffleV3_Common< GridwiseGemm, AsDataType, BsDataType, DsDataType, EDataType, MPerBlock, NPerBlock, KPerBlock, BlockSize, AK1, BK1, GemmSpec, CDEShuffleBlockTransferScalarPerVectors, BlkGemmPipeSched, BlkGemmPipelineVer, ComputeTypeA, ComputeTypeB >::Argument = typename GridwiseGemm::Argument

Member Function Documentation

◆ IsSupportedArgument()

template<typename GridwiseGemm, typename AsDataType, typename BsDataType, typename DsDataType, typename EDataType, index_t MPerBlock, index_t NPerBlock, index_t KPerBlock, index_t BlockSize, index_t AK1, index_t BK1, GemmSpecialization GemmSpec, typename CDEShuffleBlockTransferScalarPerVectors, BlockGemmPipelineScheduler BlkGemmPipeSched, BlockGemmPipelineVersion BlkGemmPipelineVer, typename ComputeTypeA, typename ComputeTypeB>

bool ck::tensor_operation::device::DeviceGemm_Wmma_CShuffleV3_Common< GridwiseGemm, AsDataType, BsDataType, DsDataType, EDataType, MPerBlock, NPerBlock, KPerBlock, BlockSize, AK1, BK1, GemmSpec, CDEShuffleBlockTransferScalarPerVectors, BlkGemmPipeSched, BlkGemmPipelineVer, ComputeTypeA, ComputeTypeB >::IsSupportedArgument ( const Argument & arg )

inlinestatic

◆ IsValidCompilationParameter()

template<typename GridwiseGemm, typename AsDataType, typename BsDataType, typename DsDataType, typename EDataType, index_t MPerBlock, index_t NPerBlock, index_t KPerBlock, index_t BlockSize, index_t AK1, index_t BK1, GemmSpecialization GemmSpec, typename CDEShuffleBlockTransferScalarPerVectors, BlockGemmPipelineScheduler BlkGemmPipeSched, BlockGemmPipelineVersion BlkGemmPipelineVer, typename ComputeTypeA, typename ComputeTypeB>

constexpr bool ck::tensor_operation::device::DeviceGemm_Wmma_CShuffleV3_Common< GridwiseGemm, AsDataType, BsDataType, DsDataType, EDataType, MPerBlock, NPerBlock, KPerBlock, BlockSize, AK1, BK1, GemmSpec, CDEShuffleBlockTransferScalarPerVectors, BlkGemmPipeSched, BlkGemmPipelineVer, ComputeTypeA, ComputeTypeB >::IsValidCompilationParameter ( )

inlinestaticconstexpr

---

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

• device_gemm_wmma_cshuffle_v3_common.hpp