Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ > Struct Template Reference

Rmsnorm2dFwdPipelineProblem&lt; XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ &gt; Struct Template Reference#

Composable Kernel: ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ > Struct Template Reference
Public Types | Static Public Attributes | List of all members
ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ > Struct Template Reference

#include <rmsnorm2d_fwd_pipeline_problem.hpp>

Public Types

using XDataType = remove_cvref_t<XDataType_>
using GammaDataType = remove_cvref_t<GammaDataType_>
using ComputeDataType = remove_cvref_t<ComputeDataType_>
using YDataType = remove_cvref_t<YDataType_>
using InvRmsDataType = remove_cvref_t<InvRmsDataType_>
using UnquantYDataType = remove_cvref_t<UnquantYDataType_>
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

◆ BlockShape

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::BlockShape = remove_cvref_t<BlockShape_>

◆ ComputeDataType

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::ComputeDataType = remove_cvref_t<ComputeDataType_>

◆ GammaDataType

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::GammaDataType = remove_cvref_t<GammaDataType_>

◆ InvRmsDataType

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::InvRmsDataType = remove_cvref_t<InvRmsDataType_>

◆ SmoothScaleDataType

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::SmoothScaleDataType = remove_cvref_t<SmoothScaleDataType_>

◆ Traits

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::Traits = remove_cvref_t<Traits_>

◆ UnquantYDataType

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::UnquantYDataType = remove_cvref_t<UnquantYDataType_>

◆ XDataType

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::XDataType = remove_cvref_t<XDataType_>

◆ YDataType

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::YDataType = remove_cvref_t<YDataType_>

◆ YScaleDataType

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
using ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::YScaleDataType = remove_cvref_t<YScaleDataType_>

Member Data Documentation

◆ kNeedCrossLaneSync

template<typename XDataType_, typename GammaDataType_, typename ComputeDataType_, typename YDataType_, typename InvRmsDataType_, typename UnquantYDataType_, typename SmoothScaleDataType_, typename YScaleDataType_, typename BlockShape_, typename Traits_>
bool ck_tile::Rmsnorm2dFwdPipelineProblem< XDataType_, GammaDataType_, ComputeDataType_, YDataType_, InvRmsDataType_, UnquantYDataType_, SmoothScaleDataType_, YScaleDataType_, BlockShape_, Traits_ >::kNeedCrossLaneSync = BlockShape::ThreadPerWarp_N > 1
staticconstexpr

◆ kNeedCrossWarpSync

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