GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ > Struct Template Reference#
Classes |
Public Types |
Public Member Functions |
Static Public Member Functions |
Static Public Attributes |
List of all members
ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ > Struct Template Reference
#include <grouped_gemm_kernel.hpp>
Inheritance diagram for ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >:

Classes | |
struct | GemmTransKernelArg |
Public Types | |
using | TilePartitioner = remove_cvref_t< TilePartitioner_ > |
using | GemmPipeline = remove_cvref_t< GemmPipeline_ > |
using | EpiloguePipeline = remove_cvref_t< EpiloguePipeline_ > |
using | ALayout = remove_cvref_t< typename GemmPipeline::ALayout > |
using | BLayout = remove_cvref_t< typename GemmPipeline::BLayout > |
using | CLayout = remove_cvref_t< typename GemmPipeline::CLayout > |
using | ADataType = remove_cvref_t< typename GemmPipeline::ADataType > |
using | BDataType = remove_cvref_t< typename GemmPipeline::BDataType > |
using | CDataType = remove_cvref_t< typename EpiloguePipeline::ODataType > |
using | OffsetTile1DPartitioner = OffsettedTile1DPartitioner< TilePartitioner > |
using | Base = GemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ > |
using | GemmKernelArgs = typename Base::GemmKernelArgs |
![]() | |
using | TilePartitioner = remove_cvref_t< TilePartitioner_ > |
using | GemmPipeline = remove_cvref_t< GemmPipeline_ > |
using | EpiloguePipeline = remove_cvref_t< EpiloguePipeline_ > |
using | ALayout = remove_cvref_t< typename GemmPipeline::ALayout > |
using | BLayout = remove_cvref_t< typename GemmPipeline::BLayout > |
using | CLayout = remove_cvref_t< typename GemmPipeline::CLayout > |
using | ADataType = remove_cvref_t< typename GemmPipeline::ADataType > |
using | BDataType = remove_cvref_t< typename GemmPipeline::BDataType > |
using | CDataType = remove_cvref_t< typename EpiloguePipeline::ODataType > |
Public Member Functions | |
CK_TILE_DEVICE void | Run (const GemmTransKernelArg &kargs) const |
CK_TILE_DEVICE void | operator() (const void CK_CONSTANT_ADDRESS_SPACE *gemm_descs_const, index_t group_count) const |
![]() | |
CK_TILE_DEVICE void | operator() (GemmKernelArgs kargs) const |
Static Public Member Functions | |
static __host__ auto | GetWorkSpaceSize (const std::vector< GroupedGemmHostArgs > &gemm_descs) -> std::size_t |
static constexpr __host__ auto | BlockSize () -> dim3 |
static constexpr __host__ auto | GridSize (const std::vector< GroupedGemmHostArgs > &gemm_descs) |
static CK_TILE_HOST auto | MakeKargs (const std::vector< GroupedGemmHostArgs > &gemm_descs) -> std::vector< GemmTransKernelArg > |
static constexpr CK_TILE_HOST_DEVICE auto | GetSmemSize () -> index_t |
![]() | |
static constexpr CK_TILE_HOST auto | GridSize (index_t M, index_t N, index_t KBatch) |
static constexpr CK_TILE_HOST auto | BlockSize () |
static constexpr CK_TILE_HOST GemmKernelArgs | MakeKernelArgs (const GemmHostArgs &hostArgs) |
static constexpr CK_TILE_HOST_DEVICE index_t | GetSmemSize () |
static CK_TILE_HOST bool | IsSupportedArgument (const GemmKernelArgs &kargs) |
template<memory_operation_enum DstInMemOp = memory_operation_enum::set> | |
static CK_TILE_DEVICE auto | MakeGemmTensorViews (const ADataType *a_ptr, const BDataType *b_ptr, CDataType *c_ptr, const GemmKernelArgs &kargs, const SplitKBatchOffset &splitk_batch_offset) |
template<typename TensorView > | |
static CK_TILE_DEVICE auto | MakeGemmPadViews (const TensorView &views) |
template<typename PadView > | |
static CK_TILE_DEVICE auto | MakeGemmTileWindows (const PadView &views, const index_t i_m, const index_t i_n) |
template<memory_operation_enum DstInMemOp = memory_operation_enum::set> | |
static CK_TILE_DEVICE void | RunGemm (const ADataType *a_ptr, const BDataType *b_ptr, CDataType *c_ptr, void *smem_ptr, const GemmKernelArgs &kargs, const SplitKBatchOffset &splitk_batch_offset, const index_t block_idx_m, const index_t block_idx_n) |
Runs single GEMM problem cooperatively by whole workgroup. More... | |
Static Public Attributes | |
static constexpr index_t | KernelBlockSize = GemmPipeline::BlockSize |
![]() | |
static constexpr index_t | KernelBlockSize = GemmPipeline::BlockSize |
static constexpr auto | I0 = number<0>() |
static constexpr auto | I1 = number<1>() |
static constexpr auto | I2 = number<2>() |
Member Typedef Documentation
◆ ADataType
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::ADataType = remove_cvref_t<typename GemmPipeline::ADataType> |
◆ ALayout
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::ALayout = remove_cvref_t<typename GemmPipeline::ALayout> |
◆ Base
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::Base = GemmKernel<TilePartitioner_, GemmPipeline_, EpiloguePipeline_> |
◆ BDataType
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::BDataType = remove_cvref_t<typename GemmPipeline::BDataType> |
◆ BLayout
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::BLayout = remove_cvref_t<typename GemmPipeline::BLayout> |
◆ CDataType
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::CDataType = remove_cvref_t<typename EpiloguePipeline::ODataType> |
◆ CLayout
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::CLayout = remove_cvref_t<typename GemmPipeline::CLayout> |
◆ EpiloguePipeline
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::EpiloguePipeline = remove_cvref_t<EpiloguePipeline_> |
◆ GemmKernelArgs
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::GemmKernelArgs = typename Base::GemmKernelArgs |
◆ GemmPipeline
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::GemmPipeline = remove_cvref_t<GemmPipeline_> |
◆ OffsetTile1DPartitioner
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::OffsetTile1DPartitioner = OffsettedTile1DPartitioner<TilePartitioner> |
◆ TilePartitioner
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
using ck_tile::GroupedGemmKernel< TilePartitioner_, GemmPipeline_, EpiloguePipeline_ >::TilePartitioner = remove_cvref_t<TilePartitioner_> |
Member Function Documentation
◆ BlockSize()
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
|
inlinestaticconstexpr |
◆ GetSmemSize()
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
|
inlinestaticconstexpr |
◆ GetWorkSpaceSize()
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
|
inlinestatic |
◆ GridSize()
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
|
inlinestaticconstexpr |
◆ MakeKargs()
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
|
inlinestatic |
◆ operator()()
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
|
inline |
◆ Run()
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
|
inline |
Member Data Documentation
◆ KernelBlockSize
template<typename TilePartitioner_ , typename GemmPipeline_ , typename EpiloguePipeline_ >
|
staticconstexpr |
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-6.4.3/include/ck_tile/ops/gemm/kernel/grouped_gemm_kernel.hpp