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

• ck_tile
• TileFmhaTraits

Static Public Attributes | List of all members

ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ > 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 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_

Member Data Documentation

◆ BiasEnum

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr auto ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::BiasEnum = BiasEnum_

staticconstexpr

◆ kBlockPerCu

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr index_t ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kBlockPerCu = kBlockPerCu_

staticconstexpr

◆ kDoFp8StaticQuant

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kDoFp8StaticQuant = kDoFp8StaticQuant_

staticconstexpr

◆ kHasBiasGrad

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kHasBiasGrad = kHasBiasGrad_

staticconstexpr

◆ kHasDropout

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kHasDropout = kHasDropout_

staticconstexpr

◆ kPadHeadDimQ

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kPadHeadDimQ = kPadHeadDimQ_

staticconstexpr

◆ kPadHeadDimV

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kPadHeadDimV = kPadHeadDimV_

staticconstexpr

◆ kPadSeqLenK

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kPadSeqLenK = kPadSeqLenK_

staticconstexpr

◆ kPadSeqLenQ

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kPadSeqLenQ = kPadSeqLenQ_

staticconstexpr

◆ kStoreLSE

template<bool kPadSeqLenQ_, bool kPadSeqLenK_, bool kPadHeadDimQ_, bool kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, bool kStoreLSE_, bool kHasDropout_, bool kDoFp8StaticQuant_, index_t kBlockPerCu_ = -1>

constexpr bool ck_tile::TileFmhaTraits< kPadSeqLenQ_, kPadSeqLenK_, kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kStoreLSE_, kHasDropout_, kDoFp8StaticQuant_, kBlockPerCu_ >::kStoreLSE = kStoreLSE_

staticconstexpr

---

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

• include/ck_tile/ops/fmha/pipeline/tile_fmha_traits.hpp