CMakeLists.txt 2.67 KB
Newer Older
1
add_custom_target(example_gemm_dl)
2

3
add_example_executable(example_gemm_dl_fp32 gemm_dl_fp32.cpp)
4
5
add_example_dependencies(example_gemm_dl example_gemm_dl_fp32)

6
add_example_executable(example_gemm_dl_fp16 gemm_dl_fp16.cpp)
7
8
add_example_dependencies(example_gemm_dl example_gemm_dl_fp16)

9
add_example_executable(example_gemm_dpp_fp16 gemm_dpp_fp16.cpp)
10

11
add_example_executable(example_gemm_dl_int8 gemm_dl_int8.cpp)
12
add_example_dependencies(example_gemm_dl example_gemm_dl_int8)
13
if(USE_BITINT_EXTENSION_INT4)
14
    add_example_executable(example_gemm_dl_int4 gemm_dl_int4.cpp)
15
    add_example_dependencies(example_gemm_dl example_gemm_dl_int4)
16
endif(USE_BITINT_EXTENSION_INT4)
17
18

add_custom_target(example_gemm_xdl)
19
add_example_executable(example_gemm_xdl_fp16 gemm_xdl_fp16.cpp)
20
21
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp16)

22
add_example_executable(example_gemm_xdl_wavelet_fp16 gemm_xdl_wavelet_fp16.cpp)
23
24
add_example_dependencies(example_gemm_xdl example_gemm_xdl_wavelet_fp16)

25
add_example_executable(example_gemm_xdl_skip_b_lds_fp16 gemm_xdl_skip_b_lds_fp16.cpp)
26
add_example_dependencies(example_gemm_xdl example_gemm_xdl_skip_b_lds_fp16)
27
if(GPU_TARGETS MATCHES "gfx1100" OR GPU_TARGETS MATCHES "gfx1101" OR GPU_TARGETS MATCHES "gfx1102")
28
29
    add_custom_target(example_gemm_wmma)
    add_example_executable(example_gemm_wmma_fp16 gemm_wmma_fp16.cpp)
30
    add_example_dependencies(example_gemm_wmma example_gemm_wmma_fp16)
31
32
endif()

33
add_example_executable(example_gemm_xdl_bf16 gemm_xdl_bf16.cpp)
34
add_example_dependencies(example_gemm_xdl example_gemm_xdl_bf16)
35

36
37
add_example_executable(example_gemm_xdl_bf16_rtn gemm_xdl_bf16_rtn.cpp)
add_example_dependencies(example_gemm_xdl example_gemm_xdl_bf16_rtn)
38

39
add_example_executable(example_gemm_xdl_int8 gemm_xdl_int8.cpp)
40
add_example_dependencies(example_gemm_xdl example_gemm_xdl_int8)
41

42
if(USE_BITINT_EXTENSION_INT4)
43
44
    add_example_executable(example_gemm_xdl_int4 gemm_xdl_int4.cpp)
    add_example_dependencies(example_gemm_xdl example_gemm_xdl_int4)
45
46
endif(USE_BITINT_EXTENSION_INT4)

47
48
# FIXME: re-enable this exampe as test when SWDEV-335738 is fixed
add_example_executable_no_testing(example_gemm_xdl_fp64 gemm_xdl_fp64.cpp)
49
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp64)
50

51
52
add_example_executable(example_gemm_xdl_streamk gemm_xdl_streamk.cpp)

zjing14's avatar
zjing14 committed
53
add_example_executable(example_gemm_xdl_fp8 gemm_xdl_fp8.cpp)
54
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp8)
55

zjing14's avatar
zjing14 committed
56
add_example_executable(example_gemm_xdl_fp8_bf8 gemm_xdl_fp8_bf8.cpp)
57
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp8_bf8)
58
59

add_example_executable(example_gemm_xdl_fp16_fp8 gemm_xdl_fp16_fp8.cpp)
60
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp16_fp8)