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

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

Composable Kernel: ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ > Struct Template Reference
Static Public Attributes | List of all members
ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout = kHasDropout_
 
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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr auto ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr index_t ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kHasBiasGrad = kHasBiasGrad_
staticconstexpr

◆ kHasDropout

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kHasDropout = kHasDropout_
staticconstexpr

◆ kHasLogitsSoftCap

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

◆ kPadHeadDimQ

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, bool kHasLogitsSoftCap_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, 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 kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1, bool kSkipMinSeqlenQ_ = false>
constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, kHasLogitsSoftCap_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_, kSkipMinSeqlenQ_ >::kStoreLSE = kStoreLSE_
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/fmha/pipeline/tile_fmha_traits.hpp