/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/layernorm2d/pipeline/layernorm2d_fwd_pipeline_problem.hpp Source File

• develop
• include
• ck_tile
• ops
• layernorm2d
• pipeline

layernorm2d_fwd_pipeline_problem.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

2 // Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.

3 

4 #pragma once

5 

6 #include "ck_tile/core/utility/type_traits.hpp"

7 

8 namespace ck_tile {

9 

10 template <typename XDataType_,

11  typename XBiasDataType_,

12  typename GammaDataType_,

13  typename BetaDataType_,

14  typename ComputeDataType_,

15  typename YDataType_,

16  typename MeanDataType_,

17  typename InvStdDataType_,

18  typename SmoothScaleDataType_,

19  typename YScaleDataType_,

20  typename BlockShape_,

21  typename Traits_>

22 struct Layernorm2dFwdPipelineProblem

23 {

24  using XDataType = remove_cvref_t<XDataType_>;

25  using XBiasDataType = remove_cvref_t<XBiasDataType_>;

26  using GammaDataType = remove_cvref_t<GammaDataType_>;

27  using BetaDataType = remove_cvref_t<BetaDataType_>;

28  using ComputeDataType = remove_cvref_t<ComputeDataType_>;

29  using YDataType = remove_cvref_t<YDataType_>;

30  using MeanDataType = remove_cvref_t<MeanDataType_>;

31  using InvStdDataType = remove_cvref_t<InvStdDataType_>;

32  using SmoothScaleDataType = remove_cvref_t<SmoothScaleDataType_>;

33  using YScaleDataType = remove_cvref_t<YScaleDataType_>;

34  using BlockShape = remove_cvref_t<BlockShape_>;

35 

36  static constexpr bool kNeedCrossLaneSync = BlockShape::ThreadPerWarp_N > 1;

37  static constexpr bool kNeedCrossWarpSync = BlockShape::WarpPerBlock_N > 1;

38 

39  using Traits = remove_cvref_t<Traits_>;

40 };

41 

42 } // namespace ck_tile

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::remove_cvref_t

remove_cv_t< std::remove_reference_t< T > > remove_cvref_t

Definition: type_traits.hpp:21

ck_tile::Layernorm2dFwdPipelineProblem

Definition: layernorm2d_fwd_pipeline_problem.hpp:23

ck_tile::Layernorm2dFwdPipelineProblem::XDataType

remove_cvref_t< XDataType_ > XDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:24

ck_tile::Layernorm2dFwdPipelineProblem::BlockShape

remove_cvref_t< BlockShape_ > BlockShape

Definition: layernorm2d_fwd_pipeline_problem.hpp:34

ck_tile::Layernorm2dFwdPipelineProblem::Traits

remove_cvref_t< Traits_ > Traits

Definition: layernorm2d_fwd_pipeline_problem.hpp:39

ck_tile::Layernorm2dFwdPipelineProblem::kNeedCrossLaneSync

static constexpr bool kNeedCrossLaneSync

Definition: layernorm2d_fwd_pipeline_problem.hpp:36

ck_tile::Layernorm2dFwdPipelineProblem::GammaDataType

remove_cvref_t< GammaDataType_ > GammaDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:26

ck_tile::Layernorm2dFwdPipelineProblem::InvStdDataType

remove_cvref_t< InvStdDataType_ > InvStdDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:31

ck_tile::Layernorm2dFwdPipelineProblem::ComputeDataType

remove_cvref_t< ComputeDataType_ > ComputeDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:28

ck_tile::Layernorm2dFwdPipelineProblem::kNeedCrossWarpSync

static constexpr bool kNeedCrossWarpSync

Definition: layernorm2d_fwd_pipeline_problem.hpp:37

ck_tile::Layernorm2dFwdPipelineProblem::MeanDataType

remove_cvref_t< MeanDataType_ > MeanDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:30

ck_tile::Layernorm2dFwdPipelineProblem::YScaleDataType

remove_cvref_t< YScaleDataType_ > YScaleDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:33

ck_tile::Layernorm2dFwdPipelineProblem::BetaDataType

remove_cvref_t< BetaDataType_ > BetaDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:27

ck_tile::Layernorm2dFwdPipelineProblem::XBiasDataType

remove_cvref_t< XBiasDataType_ > XBiasDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:25

ck_tile::Layernorm2dFwdPipelineProblem::YDataType

remove_cvref_t< YDataType_ > YDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:29

ck_tile::Layernorm2dFwdPipelineProblem::SmoothScaleDataType

remove_cvref_t< SmoothScaleDataType_ > SmoothScaleDataType

Definition: layernorm2d_fwd_pipeline_problem.hpp:32

type_traits.hpp