/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

• develop
• include
• ck_tile
• ops
• gemm
• warp

warp_wmma_gemm.hpp

Go to the documentation of this file.

1 #pragma once

2 

3 #include "ck_tile/core.hpp"

4 #include "ck_tile/ops/gemm/warp/warp_gemm_impl.hpp"

5 #include "ck_tile/ops/gemm/warp/warp_gemm_attribute_wmma.hpp"

6 

7 namespace ck_tile {

8 

9 template <bool kTransC = false>

10 using WarpGemmWmma_f32_16x16x16_f16_f16 =

11  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f16_f16, kTransC>>;

12 

13 template <bool kTransC = false>

14 using WarpGemmWmma_f32_16x16x16_bf16_bf16 =

15  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf16_bf16, kTransC>>;

16 

17 template <bool kTransC = false>

18 using WarpGemmWmma_i32_16x16x16_i8_i8 =

19  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_i32_16x16x16_i8_i8, kTransC>>;

20 

21 template <bool kTransC = false>

22 using WarpGemmWmma_f32_16x16x16_f8_f8 =

23  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_f8, kTransC>>;

24 

25 template <bool kTransC = false>

26 using WarpGemmWmma_f32_16x16x16_bf8_bf8 =

27  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_bf8, kTransC>>;

28 

29 template <bool kTransC = false>

30 using WarpGemmWmma_f32_16x16x16_f8_bf8 =

31  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_bf8, kTransC>>;

32 

33 template <bool kTransC = false>

34 using WarpGemmWmma_f32_16x16x16_bf8_f8 =

35  WarpGemmImpl<WarpGemmAttributeWmma<WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_f8, kTransC>>;

36 

37 } // namespace ck_tile

core.hpp

ck_tile

Definition: cluster_descriptor.hpp:13

ck_tile::WarpGemmImpl

Definition: warp_gemm_impl.hpp:11

warp_gemm_attribute_wmma.hpp

warp_gemm_impl.hpp