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

• develop
• include
• ck_tile
• ops
• reduce
• pipeline

reduce2d_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.hpp"

7 

8 namespace ck_tile {

9 

10 template <typename XDataType_,

11  typename ComputeDataType_,

12  typename YDataType_,

13  typename BlockShape_,

14  typename ReduceOp_>

15 struct Reduce2dProblem

16 {

17  using XDataType = remove_cvref_t<XDataType_>;

18  using ComputeDataType = remove_cvref_t<ComputeDataType_>;

19  using YDataType = remove_cvref_t<YDataType_>;

20  using BlockShape = remove_cvref_t<BlockShape_>;

21  using ReduceOp = ReduceOp_;

22 

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

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

25 };

26 

27 } // namespace ck_tile

core.hpp

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::Reduce2dProblem

Definition: reduce2d_problem.hpp:16

ck_tile::Reduce2dProblem::YDataType

remove_cvref_t< YDataType_ > YDataType

Definition: reduce2d_problem.hpp:19

ck_tile::Reduce2dProblem::kNeedCrossWarpSync

static constexpr bool kNeedCrossWarpSync

Definition: reduce2d_problem.hpp:24

ck_tile::Reduce2dProblem::ReduceOp

ReduceOp_ ReduceOp

Definition: reduce2d_problem.hpp:21

ck_tile::Reduce2dProblem::ComputeDataType

remove_cvref_t< ComputeDataType_ > ComputeDataType

Definition: reduce2d_problem.hpp:18

ck_tile::Reduce2dProblem::kNeedCrossLaneSync

static constexpr bool kNeedCrossLaneSync

Definition: reduce2d_problem.hpp:23

ck_tile::Reduce2dProblem::BlockShape

remove_cvref_t< BlockShape_ > BlockShape

Definition: reduce2d_problem.hpp:20

ck_tile::Reduce2dProblem::XDataType

remove_cvref_t< XDataType_ > XDataType

Definition: reduce2d_problem.hpp:17