FmhaFwdAppendKVTilePartitioner< kM0_, kN0_, kK0_, kN1_ > Struct Template Reference

• ck_tile
• FmhaFwdAppendKVTilePartitioner

Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members

ck_tile::FmhaFwdAppendKVTilePartitioner< kM0_, kN0_, kK0_, kN1_ > Struct Template Reference

#include <fmha_fwd_appendkv_tile_partitioner.hpp>

Public Member Functions
CK_TILE_DEVICE auto	operator() ()

Static Public Member Functions
static CK_TILE_HOST constexpr auto	GridSize (ck_tile::index_t batch_size, ck_tile::index_t nhead, ck_tile::index_t seqlen_q, ck_tile::index_t seqlen_knew)

Static Public Attributes
static constexpr ck_tile::index_t	kM0 = kM0_
static constexpr ck_tile::index_t	kN0 = kN0_
static constexpr ck_tile::index_t	kK0 = kK0_
static constexpr ck_tile::index_t	kN1 = kN1_

Member Function Documentation

◆ GridSize()

template<index_t kM0_, index_t kN0_, index_t kK0_, index_t kN1_>

CK_TILE_HOST constexpr auto ck_tile::FmhaFwdAppendKVTilePartitioner< kM0_, kN0_, kK0_, kN1_ >::GridSize ( ck_tile::index_t batch_size, ck_tile::index_t nhead, ck_tile::index_t seqlen_q, ck_tile::index_t seqlen_knew )

inlinestaticconstexpr

◆ operator()()

template<index_t kM0_, index_t kN0_, index_t kK0_, index_t kN1_>

CK_TILE_DEVICE auto ck_tile::FmhaFwdAppendKVTilePartitioner< kM0_, kN0_, kK0_, kN1_ >::operator() ( )

inline

Member Data Documentation

◆ kK0

template<index_t kM0_, index_t kN0_, index_t kK0_, index_t kN1_>

ck_tile::index_t ck_tile::FmhaFwdAppendKVTilePartitioner< kM0_, kN0_, kK0_, kN1_ >::kK0 = kK0_

staticconstexpr

◆ kM0

template<index_t kM0_, index_t kN0_, index_t kK0_, index_t kN1_>

ck_tile::index_t ck_tile::FmhaFwdAppendKVTilePartitioner< kM0_, kN0_, kK0_, kN1_ >::kM0 = kM0_

staticconstexpr

◆ kN0

template<index_t kM0_, index_t kN0_, index_t kK0_, index_t kN1_>

ck_tile::index_t ck_tile::FmhaFwdAppendKVTilePartitioner< kM0_, kN0_, kK0_, kN1_ >::kN0 = kN0_

staticconstexpr

◆ kN1

template<index_t kM0_, index_t kN0_, index_t kK0_, index_t kN1_>

ck_tile::index_t ck_tile::FmhaFwdAppendKVTilePartitioner< kM0_, kN0_, kK0_, kN1_ >::kN1 = kN1_

staticconstexpr

---

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

• fmha_fwd_appendkv_tile_partitioner.hpp