TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ > Struct Template Reference

TileFmhaFwdPagedKVTraits&lt; kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ > Struct Template Reference
Static Public Attributes | List of all members
ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ > Struct Template Reference

#include <tile_fmha_traits.hpp>

Static Public Attributes

static constexpr bool kPadSeqLenQ = kPadSeqLenQ_
static constexpr bool kPadSeqLenK = kPadSeqLenK_
static constexpr bool kPadHeadDimQ = kPadHeadDimQ_
static constexpr bool kPadHeadDimV = kPadHeadDimV_
static constexpr bool kHasLogitsSoftCap = kHasLogitsSoftCap_
static constexpr auto BiasEnum = BiasEnum_
static constexpr bool kHasBiasGrad = kHasBiasGrad_
static constexpr bool kStoreLSE = kStoreLSE_
static constexpr bool kIsPagedKV = kIsPagedKV_
static constexpr bool kDoFp8StaticQuant = kDoFp8StaticQuant_
static constexpr index_t kBlockPerCu = kBlockPerCu_
static constexpr bool kSkipMinSeqlenQ = kSkipMinSeqlenQ_

Member Data Documentation

◆ BiasEnum

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
auto ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::BiasEnum = BiasEnum_
staticconstexpr

◆ kBlockPerCu

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
index_t ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kBlockPerCu = kBlockPerCu_
staticconstexpr

◆ kDoFp8StaticQuant

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kDoFp8StaticQuant = kDoFp8StaticQuant_
staticconstexpr

◆ kHasBiasGrad

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kHasBiasGrad = kHasBiasGrad_
staticconstexpr

◆ kHasLogitsSoftCap

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kHasLogitsSoftCap = kHasLogitsSoftCap_
staticconstexpr

◆ kIsPagedKV

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kIsPagedKV = kIsPagedKV_
staticconstexpr

◆ kPadHeadDimQ

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kPadHeadDimQ = kPadHeadDimQ_
staticconstexpr

◆ kPadHeadDimV

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kPadHeadDimV = kPadHeadDimV_
staticconstexpr

◆ kPadSeqLenK

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kPadSeqLenK = kPadSeqLenK_
staticconstexpr

◆ kPadSeqLenQ

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kPadSeqLenQ = kPadSeqLenQ_
staticconstexpr

◆ kSkipMinSeqlenQ

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kSkipMinSeqlenQ = kSkipMinSeqlenQ_
staticconstexpr

◆ kStoreLSE

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kIsPagedKV_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
bool ck_tile::TileFmhaFwdPagedKVTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kIsPagedKV_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kStoreLSE = kStoreLSE_
staticconstexpr
The documentation for this struct was generated from the following file: