/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/wrapper/traits/blockwise_gemm_xdl_traits.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/wrapper/traits/blockwise_gemm_xdl_traits.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/wrapper/traits/blockwise_gemm_xdl_traits.hpp Source File
blockwise_gemm_xdl_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/ck.hpp"
7 
8 // Disable from doxygen docs generation
10 namespace ck {
11 namespace wrapper {
13 
26 template <typename MPerXDLValue,
27  typename NPerXDLValue,
28  typename MXdlPerWaveValue,
29  typename NXdlPerWaveValue,
30  typename K1Value>
31 struct BlockwisGemmXdlTraits
32 {
33  static constexpr auto MPerXDL = MPerXDLValue{};
34  static constexpr auto NPerXDL = NPerXDLValue{};
35  static constexpr auto MXdlPerWave = MXdlPerWaveValue{};
36  static constexpr auto NXdlPerWave = NXdlPerWaveValue{};
37  static constexpr auto K1 = K1Value{};
38 };
39 
40 // K1 = 4
41 struct BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_4K1
42  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<4>, Number<2>, Number<4>>
43 {
44 };
45 struct BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_4K1
46  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<4>, Number<4>>
47 {
48 };
49 struct BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_4K1
50  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<2>, Number<4>>
51 {
52 };
53 // K1 = 8
54 struct BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_8K1
55  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<4>, Number<2>, Number<8>>
56 {
57 };
58 struct BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_8K1
59  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<4>, Number<8>>
60 {
61 };
62 struct BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_8K1
63  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<2>, Number<8>>
64 {
65 };
66 // K1 = 16
67 struct BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_16K1
68  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<4>, Number<2>, Number<16>>
69 {
70 };
71 struct BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_16K1
72  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<4>, Number<16>>
73 {
74 };
75 struct BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_16K1
76  : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<2>, Number<16>>
77 {
78 };
79 
80 } // namespace wrapper
81 } // namespace ck
ck
Definition: ck.hpp:264
BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_16K1
Definition: blockwise_gemm_xdl_traits.hpp:77
BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_4K1
Definition: blockwise_gemm_xdl_traits.hpp:51
BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_8K1
Definition: blockwise_gemm_xdl_traits.hpp:64
BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_16K1
Definition: blockwise_gemm_xdl_traits.hpp:73
BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_4K1
Definition: blockwise_gemm_xdl_traits.hpp:47
BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_8K1
Definition: blockwise_gemm_xdl_traits.hpp:60
BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_16K1
Definition: blockwise_gemm_xdl_traits.hpp:69
BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_4K1
Definition: blockwise_gemm_xdl_traits.hpp:43
BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_8K1
Definition: blockwise_gemm_xdl_traits.hpp:56
BlockwisGemmXdlTraits
Traits for blockwise gemm xdl.
Definition: blockwise_gemm_xdl_traits.hpp:32