BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ > Struct Template Reference

BlockFmhaBwdConvertQGradPipelineProblem&lt; AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ > Struct Template Reference
Public Types | Static Public Attributes | List of all members
ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ > Struct Template Reference

#include <block_fmha_bwd_pipeline_problem.hpp>

Public Types

using AccDataType = remove_cvref_t<AccDataType_>
using QGradDataType = remove_cvref_t<QGradDataType_>
using Traits = remove_cvref_t<Traits_>

Static Public Attributes

static constexpr index_t kBlockSize = kBlockSize_
static constexpr index_t kM0 = kM0_
static constexpr index_t kN0 = kN0_
static constexpr index_t kQKHeaddim = kQKHeaddim_
static constexpr bool kIsGroupMode = kIsGroupMode_
static constexpr bool kIsDeterministic = kIsDeterministic_
static constexpr bool kPadSeqLenQ = Traits::kPadSeqLenQ
static constexpr bool kPadHeadDimQ = Traits::kPadHeadDimQ
static constexpr index_t kBlockPerCu = Traits::kBlockPerCu

Member Typedef Documentation

◆ AccDataType

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
using ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::AccDataType = remove_cvref_t<AccDataType_>

◆ QGradDataType

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
using ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::QGradDataType = remove_cvref_t<QGradDataType_>

◆ Traits

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
using ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::Traits = remove_cvref_t<Traits_>

Member Data Documentation

◆ kBlockPerCu

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
index_t ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kBlockPerCu = Traits::kBlockPerCu
staticconstexpr

◆ kBlockSize

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
index_t ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kBlockSize = kBlockSize_
staticconstexpr

◆ kIsDeterministic

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
bool ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kIsDeterministic = kIsDeterministic_
staticconstexpr

◆ kIsGroupMode

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
bool ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kIsGroupMode = kIsGroupMode_
staticconstexpr

◆ kM0

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
index_t ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kM0 = kM0_
staticconstexpr

◆ kN0

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
index_t ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kN0 = kN0_
staticconstexpr

◆ kPadHeadDimQ

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
bool ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kPadHeadDimQ = Traits::kPadHeadDimQ
staticconstexpr

◆ kPadSeqLenQ

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
bool ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kPadSeqLenQ = Traits::kPadSeqLenQ
staticconstexpr

◆ kQKHeaddim

template<typename AccDataType_, typename QGradDataType_, index_t kBlockSize_, index_t kM0_, index_t kN0_, index_t kQKHeaddim_, bool kIsGroupMode_, bool kIsDeterministic_, typename Traits_>
index_t ck_tile::BlockFmhaBwdConvertQGradPipelineProblem< AccDataType_, QGradDataType_, kBlockSize_, kM0_, kN0_, kQKHeaddim_, kIsGroupMode_, kIsDeterministic_, Traits_ >::kQKHeaddim = kQKHeaddim_
staticconstexpr
The documentation for this struct was generated from the following file: