TileFmhaBwdTraits< kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ > Struct Template Reference

TileFmhaBwdTraits&lt; kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::TileFmhaBwdTraits< kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ > Struct Template Reference
Static Public Attributes | List of all members
ck_tile::TileFmhaBwdTraits< kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ > Struct Template Reference

#include <tile_fmha_traits.hpp>

Static Public Attributes

static constexpr index_t kPadHeadDimQ = kPadHeadDimQ_
static constexpr index_t kPadHeadDimV = kPadHeadDimV_
static constexpr auto BiasEnum = BiasEnum_
static constexpr bool kHasBiasGrad = kHasBiasGrad_
static constexpr index_t kBlockPerCu = kBlockPerCu_

Member Data Documentation

◆ BiasEnum

template<index_t kPadHeadDimQ_, index_t kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, index_t kBlockPerCu_ = -1>
auto ck_tile::TileFmhaBwdTraits< kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ >::BiasEnum = BiasEnum_
staticconstexpr

◆ kBlockPerCu

template<index_t kPadHeadDimQ_, index_t kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, index_t kBlockPerCu_ = -1>
index_t ck_tile::TileFmhaBwdTraits< kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ >::kBlockPerCu = kBlockPerCu_
staticconstexpr

◆ kHasBiasGrad

template<index_t kPadHeadDimQ_, index_t kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, index_t kBlockPerCu_ = -1>
bool ck_tile::TileFmhaBwdTraits< kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ >::kHasBiasGrad = kHasBiasGrad_
staticconstexpr

◆ kPadHeadDimQ

template<index_t kPadHeadDimQ_, index_t kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, index_t kBlockPerCu_ = -1>
index_t ck_tile::TileFmhaBwdTraits< kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ >::kPadHeadDimQ = kPadHeadDimQ_
staticconstexpr

◆ kPadHeadDimV

template<index_t kPadHeadDimQ_, index_t kPadHeadDimV_, BlockAttentionBiasEnum BiasEnum_, bool kHasBiasGrad_, index_t kBlockPerCu_ = -1>
index_t ck_tile::TileFmhaBwdTraits< kPadHeadDimQ_, kPadHeadDimV_, BiasEnum_, kHasBiasGrad_, kBlockPerCu_ >::kPadHeadDimV = kPadHeadDimV_
staticconstexpr
The documentation for this struct was generated from the following file: