warp_wmma_gemm.hpp Source File

warp_wmma_gemm.hpp Source File#

Composable Kernel: 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
12namespace ck_tile {
13
14template <bool kTransC = false>
15using WarpGemmWmma_f32_16x16x16_f16_f16 =
16 WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f16_f16, kTransC>>;
17
18template <bool kTransC = false>
19using WarpGemmWmma_f32_16x16x16_bf16_bf16 =
20 WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf16_bf16, kTransC>>;
21
22template <bool kTransC = false>
23using WarpGemmWmma_i32_16x16x16_i8_i8 =
24 WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_i32_16x16x16_i8_i8, kTransC>>;
25
26template <bool kTransC = false>
27using WarpGemmWmma_f32_16x16x16_f8_f8 =
28 WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_f8, kTransC>>;
29
30template <bool kTransC = false>
31using WarpGemmWmma_f32_16x16x16_bf8_bf8 =
32 WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_bf8, kTransC>>;
33
34template <bool kTransC = false>
35using WarpGemmWmma_f32_16x16x16_f8_bf8 =
36 WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_bf8, kTransC>>;
37
38template <bool kTransC = false>
39using WarpGemmWmma_f32_16x16x16_bf8_f8 =
40 WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_f8, kTransC>>;
41
42} // namespace ck_tile
ck_tile
Definition tile/core/algorithm/cluster_descriptor.hpp:13
ck_tile::WarpGemmWmma_f32_16x16x16_bf16_bf16
WarpGemmImpl< WarpGemmAttributeWmma< WarpGemmAttributeWmmaImpl_f32_16x16x16_bf16_bf16, kTransC > > WarpGemmWmma_f32_16x16x16_bf16_bf16
Definition warp_wmma_gemm.hpp:19
ck_tile::WarpGemmWmma_f32_16x16x16_f8_f8
WarpGemmImpl< WarpGemmAttributeWmma< WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_f8, kTransC > > WarpGemmWmma_f32_16x16x16_f8_f8
Definition warp_wmma_gemm.hpp:27
ck_tile::WarpGemmWmma_f32_16x16x16_bf8_bf8
WarpGemmImpl< WarpGemmAttributeWmma< WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_bf8, kTransC > > WarpGemmWmma_f32_16x16x16_bf8_bf8
Definition warp_wmma_gemm.hpp:31
ck_tile::WarpGemmWmma_i32_16x16x16_i8_i8
WarpGemmImpl< WarpGemmAttributeWmma< WarpGemmAttributeWmmaImpl_i32_16x16x16_i8_i8, kTransC > > WarpGemmWmma_i32_16x16x16_i8_i8
Definition warp_wmma_gemm.hpp:23
ck_tile::WarpGemmWmma_f32_16x16x16_f16_f16
WarpGemmImpl< WarpGemmAttributeWmma< WarpGemmAttributeWmmaImpl_f32_16x16x16_f16_f16, kTransC > > WarpGemmWmma_f32_16x16x16_f16_f16
Definition warp_wmma_gemm.hpp:15
ck_tile::WarpGemmWmma_f32_16x16x16_f8_bf8
WarpGemmImpl< WarpGemmAttributeWmma< WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_bf8, kTransC > > WarpGemmWmma_f32_16x16x16_f8_bf8
Definition warp_wmma_gemm.hpp:35
ck_tile::WarpGemmWmma_f32_16x16x16_bf8_f8
WarpGemmImpl< WarpGemmAttributeWmma< WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_f8, kTransC > > WarpGemmWmma_f32_16x16x16_bf8_f8
Definition warp_wmma_gemm.hpp:39
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