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

• develop
• include
• ck_tile
• ops
• gemm
• pipeline

tile_gemm_shape.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 #include "ck_tile/host/concat.hpp"

8 

9 namespace ck_tile {

10 

11 template <typename BlockTile_,

12  typename BlockWarps_,

13  typename WarpTile_,

14  bool PermuteA_ = false,

15  bool PermuteB_ = false>

16 struct TileGemmShape

17 {

18  using BlockTile = remove_cvref_t<BlockTile_>;

19  using BlockWarps = remove_cvref_t<BlockWarps_>;

20  using WarpTile = remove_cvref_t<WarpTile_>;

21 

22  static constexpr index_t NumWarps = reduce_on_sequence(BlockWarps{}, multiplies{}, number<1>{});

23 

24  static constexpr index_t kM = BlockTile::at(number<0>{});

25  static constexpr index_t kN = BlockTile::at(number<1>{});

26  static constexpr index_t kK = BlockTile::at(number<2>{});

27 

28  static constexpr bool PermuteA = PermuteA_;

29  static constexpr bool PermuteB = PermuteB_;

30 

31  static constexpr index_t flatNPerWarp = BlockWarps::at(number<1>{});

32  static constexpr index_t flatKPerWarp = WarpTile::at(number<2>{}) * WarpTile::at(number<1>{});

33  static constexpr index_t flatKPerBlock = flatKPerWarp * kK / WarpTile::at(number<2>{});

34 

35  CK_TILE_HOST static std::string GetName()

36  {

37  // clang-format off

38  return concat('_', "tile_gemm_shape",

39  concat('x', kM, kN, kK, NumWarps),

40  concat('x', BlockWarps::at(number<0>{}), BlockWarps::at(number<1>{}), BlockWarps::at(number<2>{})),

41  concat('x', (WarpTile::at(number<0>{})), WarpTile::at(number<1>{}), WarpTile::at(number<2>{})));

42  // clang-format on

43  }

44 };

45 

46 } // namespace ck_tile

concat.hpp

CK_TILE_HOST

#define CK_TILE_HOST

Definition: config.hpp:40

core.hpp

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::index_t

int32_t index_t

Definition: integer.hpp:9

ck_tile::concat

auto concat(const Ts &... xs) -> std::enable_if_t<!AllConvertibleToStringView< Ts... >, std::string >

Definition: concat.hpp:43

ck_tile::remove_cvref_t

remove_cv_t< std::remove_reference_t< T > > remove_cvref_t

Definition: type_traits.hpp:21

ck_tile::reduce_on_sequence

constexpr CK_TILE_HOST_DEVICE index_t reduce_on_sequence(Seq, Reduce f, number< Init >)

Definition: sequence.hpp:979

ck_tile::TileGemmShape

Definition: tile_gemm_shape.hpp:17

ck_tile::TileGemmShape::flatNPerWarp

static constexpr index_t flatNPerWarp

Definition: tile_gemm_shape.hpp:31

ck_tile::TileGemmShape::kK

static constexpr index_t kK

Definition: tile_gemm_shape.hpp:26

ck_tile::TileGemmShape::PermuteB

static constexpr bool PermuteB

Definition: tile_gemm_shape.hpp:29

ck_tile::TileGemmShape::BlockWarps

remove_cvref_t< BlockWarps_ > BlockWarps

Definition: tile_gemm_shape.hpp:19

ck_tile::TileGemmShape::kN

static constexpr index_t kN

Definition: tile_gemm_shape.hpp:25

ck_tile::TileGemmShape::kM

static constexpr index_t kM

Definition: tile_gemm_shape.hpp:24

ck_tile::TileGemmShape::flatKPerWarp

static constexpr index_t flatKPerWarp

Definition: tile_gemm_shape.hpp:32

ck_tile::TileGemmShape::GetName

static CK_TILE_HOST std::string GetName()

Definition: tile_gemm_shape.hpp:35

ck_tile::TileGemmShape::flatKPerBlock

static constexpr index_t flatKPerBlock

Definition: tile_gemm_shape.hpp:33

ck_tile::TileGemmShape::PermuteA

static constexpr bool PermuteA

Definition: tile_gemm_shape.hpp:28

ck_tile::TileGemmShape::BlockTile

remove_cvref_t< BlockTile_ > BlockTile

Definition: tile_gemm_shape.hpp:18

ck_tile::TileGemmShape::WarpTile

remove_cvref_t< WarpTile_ > WarpTile

Definition: tile_gemm_shape.hpp:20

ck_tile::TileGemmShape::NumWarps

static constexpr index_t NumWarps

Definition: tile_gemm_shape.hpp:22

ck_tile::constant

Definition: integral_constant.hpp:13

ck_tile::multiplies

Definition: math.hpp:98