GemmWPQuantPipelineAgBgCrPolicy Struct Reference

• ck_tile
• GemmWPQuantPipelineAgBgCrPolicy

Static Public Member Functions | List of all members

ck_tile::GemmWPQuantPipelineAgBgCrPolicy Struct Reference

#include <gemm_wp_bquant_pipeline_ag_bg_cr_base_policy.hpp>

Inheritance diagram for ck_tile::GemmWPQuantPipelineAgBgCrPolicy:

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	GetBlockWeightPreshuffleBQuant ()
Static Public Member Functions inherited from ck_tile::UniversalWeightPreshufflePipelineAgBgCrPolicy
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto	MakeALdsBlockDescriptor ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr index_t	GetSmemSizeA ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr index_t	GetSmemSize ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto	GetSmemPackA ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto	GetKBPerLoad ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto	MakeADramTileDistribution ()
template<typename Problem>
static CK_TILE_DEVICE constexpr auto	MakeBFlatDramTileDistribution ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto	MakeShuffledARegBlockDistribution ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto	GetBlockWeightPreshuffle ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr auto	GetVectorSizeC ()
	Get the vector store size for C tensor.
Static Public Member Functions inherited from ck_tile::UniversalGemmBasePolicy< UniversalWeightPreshufflePipelineAgBgCrPolicy >
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
Public Types inherited from ck_tile::UniversalWeightPreshufflePipelineAgBgCrPolicy
using	BasePolicy = UniversalGemmBasePolicy<UniversalWeightPreshufflePipelineAgBgCrPolicy>
Static Public Attributes inherited from ck_tile::UniversalGemmBasePolicy< UniversalWeightPreshufflePipelineAgBgCrPolicy >
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 Function Documentation

◆ GetBlockWeightPreshuffleBQuant()

template<typename Problem>

CK_TILE_HOST_DEVICE constexpr auto ck_tile::GemmWPQuantPipelineAgBgCrPolicy::GetBlockWeightPreshuffleBQuant ( )

inlinestaticconstexpr

◆ GetVectorSizeBQ()

template<typename Problem>

CK_TILE_HOST_DEVICE constexpr auto ck_tile::GemmWPQuantPipelineAgBgCrPolicy::GetVectorSizeBQ ( )

inlinestaticconstexpr

◆ MakeBQDramTileDistribution()

template<typename Problem>

CK_TILE_HOST_DEVICE constexpr auto ck_tile::GemmWPQuantPipelineAgBgCrPolicy::MakeBQDramTileDistribution ( )

inlinestaticconstexpr

---

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

• gemm_wp_bquant_pipeline_ag_bg_cr_base_policy.hpp