GemmBQuantPipelineAgBgCrDefaultPolicy Struct Reference

GemmBQuantPipelineAgBgCrDefaultPolicy Struct Reference#

Composable Kernel: ck_tile::GemmBQuantPipelineAgBgCrDefaultPolicy Struct Reference
ck_tile::GemmBQuantPipelineAgBgCrDefaultPolicy Struct Reference

#include <gemm_bquant_pipeline_ag_bg_cr_policy.hpp>

Inheritance diagram for ck_tile::GemmBQuantPipelineAgBgCrDefaultPolicy:
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 GetVectorSizeBQ ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto MakeBQDramTileDistribution ()
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::GemmBQuantPipelineAgBgCrDefaultPolicy::GetBlockGemm ( )
inlinestaticconstexpr

◆ GetVectorSizeBQ()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GemmBQuantPipelineAgBgCrDefaultPolicy::GetVectorSizeBQ ( )
inlinestaticconstexpr

◆ MakeBQDramTileDistribution()

template<typename Problem>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::GemmBQuantPipelineAgBgCrDefaultPolicy::MakeBQDramTileDistribution ( )
inlinestaticconstexpr

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