BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ > Struct Template Reference

BlockFmhaBwdOGradDotOPipelineProblem&lt; ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ > Struct Template Reference
Public Types | Static Public Attributes | List of all members
ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ > Struct Template Reference

#include <block_fmha_bwd_pipeline_problem.hpp>

Public Types

using ODataType = remove_cvref_t<ODataType_>
using OGradDataType = remove_cvref_t<OGradDataType_>
using DDataType = remove_cvref_t<DDataType_>
using Traits = remove_cvref_t<Traits_>

Static Public Attributes

static constexpr index_t kBlockSize = kBlockSize_
static constexpr index_t kVHeaddim = kVHeaddim_
static constexpr bool kIsGroupMode = kIsGroupMode_
static constexpr bool kPadSeqLenQ = Traits::kPadSeqLenQ
static constexpr bool kPadHeadDimV = Traits::kPadHeadDimV
static constexpr index_t kBlockPerCu = Traits::kBlockPerCu

Member Typedef Documentation

◆ DDataType

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
using ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::DDataType = remove_cvref_t<DDataType_>

◆ ODataType

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
using ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::ODataType = remove_cvref_t<ODataType_>

◆ OGradDataType

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
using ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::OGradDataType = remove_cvref_t<OGradDataType_>

◆ Traits

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
using ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::Traits = remove_cvref_t<Traits_>

Member Data Documentation

◆ kBlockPerCu

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
index_t ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::kBlockPerCu = Traits::kBlockPerCu
staticconstexpr

◆ kBlockSize

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
index_t ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::kBlockSize = kBlockSize_
staticconstexpr

◆ kIsGroupMode

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
bool ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::kIsGroupMode = kIsGroupMode_
staticconstexpr

◆ kPadHeadDimV

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
bool ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::kPadHeadDimV = Traits::kPadHeadDimV
staticconstexpr

◆ kPadSeqLenQ

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
bool ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::kPadSeqLenQ = Traits::kPadSeqLenQ
staticconstexpr

◆ kVHeaddim

template<typename ODataType_, typename OGradDataType_, typename DDataType_, index_t kBlockSize_, index_t kVHeaddim_, bool kIsGroupMode_, typename Traits_>
index_t ck_tile::BlockFmhaBwdOGradDotOPipelineProblem< ODataType_, OGradDataType_, DDataType_, kBlockSize_, kVHeaddim_, kIsGroupMode_, Traits_ >::kVHeaddim = kVHeaddim_
staticconstexpr
The documentation for this struct was generated from the following file: