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

• docs-6.4.3
• include
• ck_tile
• ops
• gemm
• pipeline

tile_gemm_traits.hpp

Go to the documentation of this file.

1 // SPDX-License-Identifier: MIT

2 // Copyright (c) 2024, 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 <bool kPadM_,

11  bool kPadN_,

12  bool kPadK_,

13  typename ALayout_,

14  typename BLayout_,

15  typename CLayout_>

16 struct TileGemmTraits

17 {

18  static constexpr bool kPadM = kPadM_;

19  static constexpr bool kPadN = kPadN_;

20  static constexpr bool kPadK = kPadK_;

21 

22  // TODO this can't be hardcoded here! Should be in policy!

23  static constexpr int _VectorSize = 16;

24 

25  using ALayout = ALayout_;

26  using BLayout = BLayout_;

27  using CLayout = CLayout_;

28 

29  static constexpr bool TransposeC = false;

30 };

31 

32 template <bool kPadM_,

33  bool kPadN_,

34  bool kPadK_,

35  typename ALayout_,

36  typename BLayout_,

37  typename CLayout_,

38  bool TransposeC_ = false>

39 struct TileGemmUniversalTraits

40 {

41  static constexpr bool kPadM = kPadM_;

42  static constexpr bool kPadN = kPadN_;

43  static constexpr bool kPadK = kPadK_;

44 

45  using ALayout = ALayout_;

46  using BLayout = BLayout_;

47  using CLayout = CLayout_;

48 

49  static constexpr bool TransposeC = TransposeC_;

50 };

51 

52 } // namespace ck_tile

core.hpp

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::TileGemmTraits

Definition: tile_gemm_traits.hpp:17

ck_tile::TileGemmTraits::TransposeC

static constexpr bool TransposeC

Definition: tile_gemm_traits.hpp:29

ck_tile::TileGemmTraits::ALayout

ALayout_ ALayout

Definition: tile_gemm_traits.hpp:25

ck_tile::TileGemmTraits::_VectorSize

static constexpr int _VectorSize

Definition: tile_gemm_traits.hpp:23

ck_tile::TileGemmTraits::BLayout

BLayout_ BLayout

Definition: tile_gemm_traits.hpp:26

ck_tile::TileGemmTraits::CLayout

CLayout_ CLayout

Definition: tile_gemm_traits.hpp:27

ck_tile::TileGemmTraits::kPadM

static constexpr bool kPadM

Definition: tile_gemm_traits.hpp:18

ck_tile::TileGemmTraits::kPadN

static constexpr bool kPadN

Definition: tile_gemm_traits.hpp:19

ck_tile::TileGemmTraits::kPadK

static constexpr bool kPadK

Definition: tile_gemm_traits.hpp:20

ck_tile::TileGemmUniversalTraits

Definition: tile_gemm_traits.hpp:40

ck_tile::TileGemmUniversalTraits::ALayout

ALayout_ ALayout

Definition: tile_gemm_traits.hpp:45

ck_tile::TileGemmUniversalTraits::CLayout

CLayout_ CLayout

Definition: tile_gemm_traits.hpp:47

ck_tile::TileGemmUniversalTraits::kPadN

static constexpr bool kPadN

Definition: tile_gemm_traits.hpp:42

ck_tile::TileGemmUniversalTraits::kPadK

static constexpr bool kPadK

Definition: tile_gemm_traits.hpp:43

ck_tile::TileGemmUniversalTraits::BLayout

BLayout_ BLayout

Definition: tile_gemm_traits.hpp:46

ck_tile::TileGemmUniversalTraits::kPadM

static constexpr bool kPadM

Definition: tile_gemm_traits.hpp:41

ck_tile::TileGemmUniversalTraits::TransposeC

static constexpr bool TransposeC

Definition: tile_gemm_traits.hpp:49