/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_problem.hpp Source File#
block_fmha_bwd_pipeline_problem.hpp
Go to the documentation of this file.
Definition: cluster_descriptor.hpp:13
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
Definition: block_fmha_bwd_pipeline_problem.hpp:106
static constexpr index_t kBlockPerCu
Definition: block_fmha_bwd_pipeline_problem.hpp:124
static constexpr index_t kM0
Definition: block_fmha_bwd_pipeline_problem.hpp:115
static constexpr index_t kQKHeaddim
Definition: block_fmha_bwd_pipeline_problem.hpp:117
static constexpr bool kIsGroupMode
Definition: block_fmha_bwd_pipeline_problem.hpp:118
static constexpr bool kPadHeadDimQ
Definition: block_fmha_bwd_pipeline_problem.hpp:123
remove_cvref_t< QGradDataType_ > QGradDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:108
static constexpr index_t kBlockSize
Definition: block_fmha_bwd_pipeline_problem.hpp:114
remove_cvref_t< AccDataType_ > AccDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:107
static constexpr bool kPadSeqLenQ
Definition: block_fmha_bwd_pipeline_problem.hpp:122
static constexpr bool kIsDeterministic
Definition: block_fmha_bwd_pipeline_problem.hpp:119
remove_cvref_t< Traits_ > Traits
Definition: block_fmha_bwd_pipeline_problem.hpp:109
static constexpr index_t kN0
Definition: block_fmha_bwd_pipeline_problem.hpp:116
Definition: block_fmha_bwd_pipeline_problem.hpp:77
remove_cvref_t< ODataType_ > ODataType
Definition: block_fmha_bwd_pipeline_problem.hpp:78
remove_cvref_t< OGradDataType_ > OGradDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:79
static constexpr bool kPadHeadDimV
Definition: block_fmha_bwd_pipeline_problem.hpp:92
remove_cvref_t< Traits_ > Traits
Definition: block_fmha_bwd_pipeline_problem.hpp:81
static constexpr index_t kVHeaddim
Definition: block_fmha_bwd_pipeline_problem.hpp:87
static constexpr index_t kBlockPerCu
Definition: block_fmha_bwd_pipeline_problem.hpp:93
static constexpr bool kPadSeqLenQ
Definition: block_fmha_bwd_pipeline_problem.hpp:91
static constexpr bool kIsGroupMode
Definition: block_fmha_bwd_pipeline_problem.hpp:88
remove_cvref_t< DDataType_ > DDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:80
static constexpr index_t kBlockSize
Definition: block_fmha_bwd_pipeline_problem.hpp:86
Definition: block_fmha_bwd_pipeline_problem.hpp:33
remove_cvref_t< BiasGradDataType_ > BiasGradDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:48
remove_cvref_t< Traits_ > Traits
Definition: block_fmha_bwd_pipeline_problem.hpp:52
remove_cvref_t< FmhaMask_ > FmhaMask
Definition: block_fmha_bwd_pipeline_problem.hpp:50
remove_cvref_t< GemmDataType_ > GemmDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:37
remove_cvref_t< KGradDataType_ > KGradDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:46
remove_cvref_t< DDataType_ > DDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:40
remove_cvref_t< BiasDataType_ > BiasDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:41
remove_cvref_t< FmhaDropout_ > FmhaDropout
Definition: block_fmha_bwd_pipeline_problem.hpp:51
static constexpr auto BiasEnum
Definition: block_fmha_bwd_pipeline_problem.hpp:62
remove_cvref_t< QGradDataType_ > QGradDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:45
static constexpr bool kIsGroupMode
Definition: block_fmha_bwd_pipeline_problem.hpp:55
remove_cvref_t< QDataType_ > QDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:34
remove_cvref_t< KDataType_ > KDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:35
static constexpr bool kHasBiasGrad
Definition: block_fmha_bwd_pipeline_problem.hpp:63
remove_cvref_t< LSEDataType_ > LSEDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:38
static constexpr index_t kBlockSize
Definition: block_fmha_bwd_pipeline_problem.hpp:54
static constexpr bool kPadHeadDimQ
Definition: block_fmha_bwd_pipeline_problem.hpp:60
static constexpr bool kUseTrLoad
Definition: block_fmha_bwd_pipeline_problem.hpp:57
remove_cvref_t< VDataType_ > VDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:36
remove_cvref_t< OGradDataType_ > OGradDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:44
remove_cvref_t< AccDataType_ > AccDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:39
remove_cvref_t< BlockFmhaShape_ > BlockFmhaShape
Definition: block_fmha_bwd_pipeline_problem.hpp:49
static constexpr index_t kBlockPerCu
Definition: block_fmha_bwd_pipeline_problem.hpp:64
remove_cvref_t< ODataType_ > ODataType
Definition: block_fmha_bwd_pipeline_problem.hpp:43
remove_cvref_t< VGradDataType_ > VGradDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:47
static constexpr bool kPadHeadDimV
Definition: block_fmha_bwd_pipeline_problem.hpp:61
remove_cvref_t< RandValOutputDataType_ > RandValOutputDataType
Definition: block_fmha_bwd_pipeline_problem.hpp:42
static constexpr bool kIsDeterministic
Definition: block_fmha_bwd_pipeline_problem.hpp:56