BlockFmhaPipelineQSKSVSDefaultPolicy Struct Reference

• ck_tile
• BlockFmhaPipelineQSKSVSDefaultPolicy

Static Public Member Functions | List of all members

ck_tile::BlockFmhaPipelineQSKSVSDefaultPolicy Struct Reference

#include <block_fmha_pipeline_qs_ks_vs_default_policy.hpp>

Inheritance diagram for ck_tile::BlockFmhaPipelineQSKSVSDefaultPolicy:

Static Public Member Functions
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr ck_tile::index_t	GetSmemSizeK ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr ck_tile::index_t	GetSmemSizeV ()
template<typename Problem>
static CK_TILE_HOST_DEVICE constexpr ck_tile::index_t	GetSmemSize ()
Static Public Member Functions inherited from ck_tile::BlockFmhaPipelineQXKSVSCustomPolicy< false, false, 1, 1 >
static CK_TILE_HOST_DEVICE constexpr auto	GetLdsBufferSequence ()
static CK_TILE_HOST_DEVICE constexpr auto	GetSmemKPackK ()
static CK_TILE_HOST_DEVICE constexpr auto	GetAlignmentK ()
static CK_TILE_HOST_DEVICE constexpr auto	GetSmemKPackV ()
static CK_TILE_HOST_DEVICE constexpr auto	GetAlignmentV ()
static CK_TILE_HOST_DEVICE constexpr auto	GetAlignmentBias ()
static CK_TILE_HOST_DEVICE constexpr auto	GetAlignmentO ()
static CK_TILE_HOST_DEVICE constexpr auto	GetSingleSmemElementSpaceSize ()
static CK_TILE_HOST_DEVICE constexpr auto	MakeKLdsBlockDescriptor ()
static CK_TILE_HOST_DEVICE constexpr auto	MakeKLdsStoreBlockDescriptor (number< IBuf >=number< 0 >{})
static CK_TILE_HOST_DEVICE constexpr auto	MakeKLdsLoadBlockDescriptor ()
static CK_TILE_HOST_DEVICE constexpr auto	MakeVLdsBlockDescriptor ()
static CK_TILE_HOST_DEVICE constexpr ck_tile::index_t	GetSmemSizeKV ()
static CK_TILE_HOST_DEVICE constexpr ck_tile::index_t	GetSmemSize ()
static CK_TILE_HOST_DEVICE constexpr std::enable_if_t< std::is_convertible_v< decltype(Problem::kHasDropout), bool >, ck_tile::index_t >	GetSmemSizeDropout (int)
static CK_TILE_HOST_DEVICE constexpr auto	MakeKDramTileDistribution ()
static CK_TILE_DEVICE constexpr auto	MakeVDramTileDistribution ()
static CK_TILE_HOST_DEVICE constexpr auto	MakeBiasDramTileDistribution ()
static CK_TILE_HOST_DEVICE constexpr auto	MakeShuffledVRegBlockDescriptor ()
static CK_TILE_HOST_DEVICE constexpr auto	GetKVBlockGemm ()

Additional Inherited Members
Public Types inherited from ck_tile::BlockFmhaPipelineQXKSVSCustomPolicy< false, false, 1, 1 >
using	QXPolicy
Static Public Attributes inherited from ck_tile::BlockFmhaPipelineQXKSVSCustomPolicy< false, false, 1, 1 >
static constexpr bool	AsyncCopy
static constexpr index_t	NumPrefetchK
static constexpr index_t	NumPrefetchV
static constexpr index_t	NumKVLdsBuffers

Member Function Documentation

◆ GetSmemSize()

template<typename Problem>

CK_TILE_HOST_DEVICE constexpr ck_tile::index_t ck_tile::BlockFmhaPipelineQSKSVSDefaultPolicy::GetSmemSize ( )

inlinestaticconstexpr

◆ GetSmemSizeK()

template<typename Problem>

CK_TILE_HOST_DEVICE constexpr ck_tile::index_t ck_tile::BlockFmhaPipelineQSKSVSDefaultPolicy::GetSmemSizeK ( )

inlinestaticconstexpr

◆ GetSmemSizeV()

template<typename Problem>

CK_TILE_HOST_DEVICE constexpr ck_tile::index_t ck_tile::BlockFmhaPipelineQSKSVSDefaultPolicy::GetSmemSizeV ( )

inlinestaticconstexpr

---

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

• block_fmha_pipeline_qs_ks_vs_default_policy.hpp