GemmAQuantPipelineAgBgCrDefaultPolicy Struct Reference

GemmAQuantPipelineAgBgCrDefaultPolicy Struct Reference#

Composable Kernel: ck_tile::GemmAQuantPipelineAgBgCrDefaultPolicy Struct Reference
ck_tile::GemmAQuantPipelineAgBgCrDefaultPolicy Struct Reference

#include <gemm_aquant_pipeline_ag_bg_cr_policy.hpp>

Inheritance diagram for ck_tile::GemmAQuantPipelineAgBgCrDefaultPolicy:
ck_tile::UniversalGemmPipelineAgBgCrPolicy ck_tile::UniversalGemmBasePolicy< UniversalGemmPipelineAgBgCrPolicy >

Public Types

using Base = UniversalGemmPipelineAgBgCrPolicy

Static Public Member Functions

template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto GetVectorSizeAQ ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto MakeAQDramTileDistribution ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto GetBlockGemm ()
Static Public Member Functions inherited from ck_tile::UniversalGemmPipelineAgBgCrPolicy
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto GetBlockGemm ()
Static Public Member Functions inherited from ck_tile::UniversalGemmBasePolicy< UniversalGemmPipelineAgBgCrPolicy >
static constexpr auto getATileAccessPattern ()
static constexpr auto getBTileAccessPattern ()
static CK_TILE_DEVICE constexpr auto MakeALdsBlockDescriptor ()
static CK_TILE_DEVICE constexpr auto MakeBLdsBlockDescriptor ()
 Create LDS block descriptor for B tensor.
static CK_TILE_HOST_DEVICE constexpr auto GetGlobalVectorLoadSize ()
 Get the maximum global memory vector load size.
static CK_TILE_HOST_DEVICE constexpr auto GetVectorSizeA ()
static CK_TILE_HOST_DEVICE constexpr auto GetVectorSizeB ()
static CK_TILE_HOST_DEVICE constexpr auto GetVectorSizeC ()
 Get the vector store size for C tensor.
static CK_TILE_HOST_DEVICE constexpr auto IsTransposeC ()
static CK_TILE_HOST_DEVICE constexpr auto MakeADramTileDistribution ()
static CK_TILE_HOST_DEVICE constexpr auto MakeBDramTileDistribution ()
static CK_TILE_HOST_DEVICE constexpr auto MakeShuffledARegTileDistribution ()
static CK_TILE_HOST_DEVICE constexpr auto MakeShuffledBRegTileDistribution ()
static CK_TILE_HOST_DEVICE constexpr auto GetSmemPackA ()
static CK_TILE_HOST_DEVICE constexpr auto GetSmemPackB ()
static CK_TILE_DEVICE constexpr index_t GetSmemSizeA ()
static CK_TILE_DEVICE constexpr index_t GetSmemSizeB ()
static CK_TILE_DEVICE constexpr index_t GetSmemSize ()

Additional Inherited Members

Static Public Attributes inherited from ck_tile::UniversalGemmBasePolicy< UniversalGemmPipelineAgBgCrPolicy >
static constexpr bool is_a_load_tr
static constexpr bool is_b_load_tr
static constexpr auto I0
static constexpr auto I1
static constexpr auto I2
static constexpr auto DefaultATileAccessPattern
static constexpr auto DefaultBTileAccessPattern

Member Typedef Documentation

◆ Base

Member Function Documentation

◆ GetBlockGemm()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GemmAQuantPipelineAgBgCrDefaultPolicy::GetBlockGemm ( )
inlinestaticconstexpr

◆ GetVectorSizeAQ()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GemmAQuantPipelineAgBgCrDefaultPolicy::GetVectorSizeAQ ( )
inlinestaticconstexpr

◆ MakeAQDramTileDistribution()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GemmAQuantPipelineAgBgCrDefaultPolicy::MakeAQDramTileDistribution ( )
inlinestaticconstexpr

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