TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ > Struct Template Reference

TileFlatmmShape&lt; BlockTile_, BlockWarps_, WarpTile_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ > Struct Template Reference
Public Types | Static Public Member Functions | Static Public Attributes | List of all members
ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ > Struct Template Reference

#include <tile_flatmm_shape.hpp>

Public Types

using BlockTile = remove_cvref_t< BlockTile_ >
 
using BlockWarps = remove_cvref_t< BlockWarps_ >
 
using WarpTile = remove_cvref_t< WarpTile_ >
 

Static Public Member Functions

static CK_TILE_HOST std::string GetName ()
 

Static Public Attributes

static constexpr auto idxM = number<0>{}
 
static constexpr auto idxN = number<1>{}
 
static constexpr auto idxK = number<2>{}
 
static constexpr index_t NumWarps = reduce_on_sequence(BlockWarps{}, multiplies{}, number<1>{})
 
static constexpr index_t kM = BlockTile::at(idxM)
 
static constexpr index_t kN = BlockTile::at(idxN)
 
static constexpr index_t kK = BlockTile::at(idxK)
 
static constexpr index_t flatNPerWarp = BlockWarps::at(idxN)
 
static constexpr index_t flatKPerWarp = WarpTile::at(idxK) * WarpTile::at(idxN)
 
static constexpr index_t flatKPerBlock = flatKPerWarp * kK / WarpTile::at(idxK)
 
static constexpr bool PermuteA = false
 
static constexpr bool PermuteB = false
 

Member Typedef Documentation

◆ BlockTile

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
using ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::BlockTile = remove_cvref_t<BlockTile_>

◆ BlockWarps

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
using ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::BlockWarps = remove_cvref_t<BlockWarps_>

◆ WarpTile

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
using ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::WarpTile = remove_cvref_t<WarpTile_>

Member Function Documentation

◆ GetName()

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
static CK_TILE_HOST std::string ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::GetName ( )
inlinestatic

Member Data Documentation

◆ flatKPerBlock

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr index_t ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::flatKPerBlock = flatKPerWarp * kK / WarpTile::at(idxK)
staticconstexpr

◆ flatKPerWarp

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr index_t ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::flatKPerWarp = WarpTile::at(idxK) * WarpTile::at(idxN)
staticconstexpr

◆ flatNPerWarp

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr index_t ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::flatNPerWarp = BlockWarps::at(idxN)
staticconstexpr

◆ idxK

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr auto ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::idxK = number<2>{}
staticconstexpr

◆ idxM

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr auto ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::idxM = number<0>{}
staticconstexpr

◆ idxN

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr auto ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::idxN = number<1>{}
staticconstexpr

◆ kK

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr index_t ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::kK = BlockTile::at(idxK)
staticconstexpr

◆ kM

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr index_t ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::kM = BlockTile::at(idxM)
staticconstexpr

◆ kN

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr index_t ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::kN = BlockTile::at(idxN)
staticconstexpr

◆ NumWarps

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr index_t ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::NumWarps = reduce_on_sequence(BlockWarps{}, multiplies{}, number<1>{})
staticconstexpr

◆ PermuteA

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr bool ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::PermuteA = false
staticconstexpr

◆ PermuteB

template<typename BlockTile_ , typename BlockWarps_ , typename WarpTile_ >
constexpr bool ck_tile::TileFlatmmShape< BlockTile_, BlockWarps_, WarpTile_ >::PermuteB = false
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/develop/include/ck_tile/ops/flatmm/pipeline/tile_flatmm_shape.hpp