CMakeLists.txt 3.31 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
23
24
add_example_executable(example_gemm_xdl_fp16_v2 gemm_xdl_fp16_v2.cpp)
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp16_v2)

25
add_example_executable(example_gemm_xdl_wavelet_fp16 gemm_xdl_wavelet_fp16.cpp)
26
27
add_example_dependencies(example_gemm_xdl example_gemm_xdl_wavelet_fp16)

28
add_example_executable(example_gemm_xdl_skip_b_lds_fp16 gemm_xdl_skip_b_lds_fp16.cpp)
29
add_example_dependencies(example_gemm_xdl example_gemm_xdl_skip_b_lds_fp16)
zjing14's avatar
zjing14 committed
30
if(GPU_TARGETS MATCHES "gfx11")
31
32
    add_custom_target(example_gemm_wmma)
    add_example_executable(example_gemm_wmma_fp16 gemm_wmma_fp16.cpp)
33
    add_example_dependencies(example_gemm_wmma example_gemm_wmma_fp16)
34
35
endif()

36
add_example_executable(example_gemm_xdl_bf16 gemm_xdl_bf16.cpp)
37
add_example_dependencies(example_gemm_xdl example_gemm_xdl_bf16)
38

39
40
add_example_executable(example_gemm_xdl_bf16_rtn gemm_xdl_bf16_rtn.cpp)
add_example_dependencies(example_gemm_xdl example_gemm_xdl_bf16_rtn)
41

42
add_example_executable(example_gemm_xdl_int8 gemm_xdl_int8.cpp)
43
add_example_dependencies(example_gemm_xdl example_gemm_xdl_int8)
44

45
if(USE_BITINT_EXTENSION_INT4)
46
47
    add_example_executable(example_gemm_xdl_int4 gemm_xdl_int4.cpp)
    add_example_dependencies(example_gemm_xdl example_gemm_xdl_int4)
48
49
endif(USE_BITINT_EXTENSION_INT4)

50
# FIXME: re-enable this example as test when SWDEV-335738 is fixed
51
add_example_executable_no_testing(example_gemm_xdl_fp64 gemm_xdl_fp64.cpp)
52
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp64)
53

54
55
add_example_executable(example_gemm_xdl_streamk gemm_xdl_streamk.cpp)

56
57
58
59
60
61
62
63
64
65
66
67
68
list(APPEND gpu_list gfx90a gfx940 gfx941 gfx942)
set(target 0)
foreach(gpu IN LISTS GPU_TARGETS)
    if(gpu IN_LIST gpu_list AND target EQUAL 0)
        add_example_executable(example_gemm_xdl_lds_direct_load_fp32 gemm_xdl_lds_direct_load_fp32.cpp)
        add_example_dependencies(example_gemm_xdl example_gemm_xdl_lds_direct_load_fp32)

        add_example_executable(example_gemm_xdl_lds_direct_load_fp16 gemm_xdl_lds_direct_load_fp16.cpp)
        add_example_dependencies(example_gemm_xdl example_gemm_xdl_lds_direct_load_fp16)
        set(target 1)
    endif()
endforeach()

zjing14's avatar
zjing14 committed
69
70
71
72
73
74
add_example_executable(example_gemm_xdl_fp8 gemm_xdl_fp8.cpp)
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp8)

add_example_executable(example_gemm_xdl_fp8_bf8 gemm_xdl_fp8_bf8.cpp)
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp8_bf8)

75
add_example_executable(example_gemm_xdl_fp16_fp8 gemm_xdl_fp16_fp8.cpp)
76
add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp16_fp8)
zjing14's avatar
zjing14 committed
77