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

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

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/gemm/warp/warp_wmma_gemm.hpp Source File
warp_wmma_gemm.hpp
Go to the documentation of this file.
1 // Copyright © Advanced Micro Devices, Inc. or its affiliates.
2 // SPDX-License-Identifier: MIT
3 
4 #pragma once
5 
6 #include "ck_tile/core.hpp"
7 #include "ck_tile/ops/gemm/warp/warp_gemm_impl.hpp"
8 #include "ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma.hpp"
9 #include "ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl_16bit_traits.hpp"
10 #include "ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma_impl_8bit_traits.hpp"
11 
12 namespace ck_tile {
13 
14 template <bool kTransC = false>
15 using WarpGemmWmma_f32_16x16x16_f16_f16 =
16  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f16_f16, kTransC>>;
17 
18 template <bool kTransC = false>
19 using WarpGemmWmma_f32_16x16x16_bf16_bf16 =
20  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf16_bf16, kTransC>>;
21 
22 template <bool kTransC = false>
23 using WarpGemmWmma_i32_16x16x16_i8_i8 =
24  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_i32_16x16x16_i8_i8, kTransC>>;
25 
26 template <bool kTransC = false>
27 using WarpGemmWmma_f32_16x16x16_f8_f8 =
28  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_f8, kTransC>>;
29 
30 template <bool kTransC = false>
31 using WarpGemmWmma_f32_16x16x16_bf8_bf8 =
32  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_bf8, kTransC>>;
33 
34 template <bool kTransC = false>
35 using WarpGemmWmma_f32_16x16x16_f8_bf8 =
36  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_bf8, kTransC>>;
37 
38 template <bool kTransC = false>
39 using WarpGemmWmma_f32_16x16x16_bf8_f8 =
40  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_f8, kTransC>>;
41 
42 } // namespace ck_tile
ck_tile
Definition: cluster_descriptor.hpp:13
ck_tile::WarpGemmImpl
Definition: warp_gemm_impl.hpp:11
warp_gemm_attribute_wmma.hpp
warp_gemm_attribute_wmma_impl_16bit_traits.hpp
warp_gemm_attribute_wmma_impl_8bit_traits.hpp
warp_gemm_impl.hpp