tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant > Struct Template Reference

tile_distribution_encoding_pattern_aq&lt; BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant &gt; Struct Template Reference#

Composable Kernel: ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant > Struct Template Reference
ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant > Struct Template Reference

#include <gemm_group_quant_utils.hpp>

Inheritance diagram for ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant >:
ck_tile::tile_distribution_encoding_pattern

Static Public Member Functions

static CK_TILE_HOST_DEVICE constexpr auto make_2d_static_tile_distribution ()

Static Public Attributes

static constexpr index_t warp_size = get_warp_size()
static constexpr index_t num_warps = BlockSize / get_warp_size()
static constexpr index_t MWarps = BlockGemmShape::BlockWarps::at(number<0>{})
static constexpr index_t NWarps = BlockGemmShape::BlockWarps::at(number<1>{})
static constexpr index_t KWarps = BlockGemmShape::BlockWarps::at(number<2>{})
static constexpr index_t MIterPerWarp = BlockGemmShape::kM / (MWarps * WarpGemm::kM)

Member Function Documentation

◆ make_2d_static_tile_distribution()

template<typename BlockGemmShape, typename WarpGemm, index_t BlockSize, index_t YPerTile, index_t XPerTile, index_t KPerBlockAQ, index_t VecSize, bool PreshuffleQuant>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant >::make_2d_static_tile_distribution ( )
inlinestaticconstexpr

Member Data Documentation

◆ KWarps

template<typename BlockGemmShape, typename WarpGemm, index_t BlockSize, index_t YPerTile, index_t XPerTile, index_t KPerBlockAQ, index_t VecSize, bool PreshuffleQuant>
index_t ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant >::KWarps = BlockGemmShape::BlockWarps::at(number<2>{})
staticconstexpr

◆ MIterPerWarp

template<typename BlockGemmShape, typename WarpGemm, index_t BlockSize, index_t YPerTile, index_t XPerTile, index_t KPerBlockAQ, index_t VecSize, bool PreshuffleQuant>
index_t ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant >::MIterPerWarp = BlockGemmShape::kM / (MWarps * WarpGemm::kM)
staticconstexpr

◆ MWarps

template<typename BlockGemmShape, typename WarpGemm, index_t BlockSize, index_t YPerTile, index_t XPerTile, index_t KPerBlockAQ, index_t VecSize, bool PreshuffleQuant>
index_t ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant >::MWarps = BlockGemmShape::BlockWarps::at(number<0>{})
staticconstexpr

◆ num_warps

template<typename BlockGemmShape, typename WarpGemm, index_t BlockSize, index_t YPerTile, index_t XPerTile, index_t KPerBlockAQ, index_t VecSize, bool PreshuffleQuant>
index_t ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant >::num_warps = BlockSize / get_warp_size()
staticconstexpr

◆ NWarps

template<typename BlockGemmShape, typename WarpGemm, index_t BlockSize, index_t YPerTile, index_t XPerTile, index_t KPerBlockAQ, index_t VecSize, bool PreshuffleQuant>
index_t ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant >::NWarps = BlockGemmShape::BlockWarps::at(number<1>{})
staticconstexpr

◆ warp_size

template<typename BlockGemmShape, typename WarpGemm, index_t BlockSize, index_t YPerTile, index_t XPerTile, index_t KPerBlockAQ, index_t VecSize, bool PreshuffleQuant>
index_t ck_tile::tile_distribution_encoding_pattern_aq< BlockGemmShape, WarpGemm, BlockSize, YPerTile, XPerTile, KPerBlockAQ, VecSize, PreshuffleQuant >::warp_size = get_warp_size()
staticconstexpr

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