Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ > Struct Template Reference

Layernorm2dFwdPipelineProblem&lt; XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ > Struct Template Reference
Public Types | Static Public Attributes | List of all members
ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ > Struct Template Reference

#include <layernorm2d_fwd_pipeline_problem.hpp>

Public Types

using XDataType = remove_cvref_t<XDataType_>
using XBiasDataType = remove_cvref_t<XBiasDataType_>
using GammaDataType = remove_cvref_t<GammaDataType_>
using BetaDataType = remove_cvref_t<BetaDataType_>
using ComputeDataType = remove_cvref_t<ComputeDataType_>
using YDataType = remove_cvref_t<YDataType_>
using MeanDataType = remove_cvref_t<MeanDataType_>
using InvStdDataType = remove_cvref_t<InvStdDataType_>
using SmoothScaleDataType = remove_cvref_t<SmoothScaleDataType_>
using YScaleDataType = remove_cvref_t<YScaleDataType_>
using BlockShape = remove_cvref_t<BlockShape_>
using Traits = remove_cvref_t<Traits_>

Static Public Attributes

static constexpr bool kNeedCrossLaneSync = BlockShape::ThreadPerWarp_N > 1
static constexpr bool kNeedCrossWarpSync = BlockShape::WarpPerBlock_N > 1

Member Typedef Documentation

◆ BetaDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::BetaDataType = remove_cvref_t<BetaDataType_>

◆ BlockShape

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::BlockShape = remove_cvref_t<BlockShape_>

◆ ComputeDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::ComputeDataType = remove_cvref_t<ComputeDataType_>

◆ GammaDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::GammaDataType = remove_cvref_t<GammaDataType_>

◆ InvStdDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::InvStdDataType = remove_cvref_t<InvStdDataType_>

◆ MeanDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::MeanDataType = remove_cvref_t<MeanDataType_>

◆ SmoothScaleDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::SmoothScaleDataType = remove_cvref_t<SmoothScaleDataType_>

◆ Traits

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::Traits = remove_cvref_t<Traits_>

◆ XBiasDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::XBiasDataType = remove_cvref_t<XBiasDataType_>

◆ XDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::XDataType = remove_cvref_t<XDataType_>

◆ YDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::YDataType = remove_cvref_t<YDataType_>

◆ YScaleDataType

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::YScaleDataType = remove_cvref_t<YScaleDataType_>

Member Data Documentation

◆ kNeedCrossLaneSync

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
bool ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::kNeedCrossLaneSync = BlockShape::ThreadPerWarp_N > 1
staticconstexpr

◆ kNeedCrossWarpSync

template<typename XDataType_, typename XBiasDataType_, typename GammaDataType_, typename BetaDataType_, typename ComputeDataType_, typename YDataType_, typename MeanDataType_, typename InvStdDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
bool ck_tile::Layernorm2dFwdPipelineProblem< XDataType_, XBiasDataType_, GammaDataType_, BetaDataType_, ComputeDataType_, YDataType_, MeanDataType_, InvStdDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::kNeedCrossWarpSync = BlockShape::WarpPerBlock_N > 1
staticconstexpr
The documentation for this struct was generated from the following file: