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

3
4
5
6
7
8
add_example_executable(example_gemm_dl_fp32 gemm_dl_fp32.cpp)
if(result EQUAL 0)
    add_dependencies(example_gemm_dl example_gemm_dl_fp32)
endif()
add_example_executable(example_gemm_dl_fp16 gemm_dl_fp16.cpp)
if(result EQUAL 0)
9
    add_dependencies(example_gemm_dl example_gemm_dl_fp16)
10
11
12
13
endif()
add_example_executable(example_gemm_dpp_fp16 gemm_dpp_fp16.cpp)
add_example_executable(example_gemm_dl_int8 gemm_dl_int8.cpp)
if(result EQUAL 0)
14
    add_dependencies(example_gemm_dl example_gemm_dl_int8)
15
16
endif()
if(USE_BITINT_EXTENSION_INT4)
17
18
    add_example_executable(example_gemm_dl_int4 gemm_dl_int4.cpp)
    add_dependencies(example_gemm_dl example_gemm_dl_int4)
19
endif(USE_BITINT_EXTENSION_INT4)
20
21

add_custom_target(example_gemm_xdl)
22
23
24
25
26
27
28
29
30
31
32
33
34
add_example_executable(example_gemm_xdl_fp16 gemm_xdl_fp16.cpp)
if(result EQUAL 0)
    add_dependencies(example_gemm_xdl example_gemm_xdl_fp16)
endif()
add_example_executable(example_gemm_xdl_wavelet_fp16 gemm_xdl_wavelet_fp16.cpp)
if(result EQUAL 0)
    add_dependencies(example_gemm_xdl example_gemm_xdl_wavelet_fp16)
endif()
add_example_executable(example_gemm_xdl_skip_b_lds_fp16 gemm_xdl_skip_b_lds_fp16.cpp)
if(result EQUAL 0)
    add_dependencies(example_gemm_xdl example_gemm_xdl_skip_b_lds_fp16)
endif()
if(GPU_TARGETS MATCHES "gfx1100" OR GPU_TARGETS MATCHES "gfx1101" OR GPU_TARGETS MATCHES "gfx1102")
35
36
    add_custom_target(example_gemm_wmma)
    add_example_executable(example_gemm_wmma_fp16 gemm_wmma_fp16.cpp)
37
38
39
    if(result EQUAL 0)
      add_dependencies(example_gemm_wmma example_gemm_wmma_fp16)
    endif()
40
41
endif()

42
43
add_example_executable(example_gemm_xdl_bf16 gemm_xdl_bf16.cpp)
if(result EQUAL 0)
44
  add_dependencies(example_gemm_xdl example_gemm_xdl_bf16)
45
46
47

  add_example_executable(example_gemm_xdl_bf16_rtn gemm_xdl_bf16_rtn.cpp)
  add_dependencies(example_gemm_xdl example_gemm_xdl_bf16_rtn)
48
endif()
49

50
51
add_example_executable(example_gemm_xdl_int8 gemm_xdl_int8.cpp)
if(result EQUAL 0)
52
53
54
  add_dependencies(example_gemm_xdl example_gemm_xdl_int8)
endif()

55
56
57
58
59
if(USE_BITINT_EXTENSION_INT4)
  add_example_executable(example_gemm_xdl_int4 gemm_xdl_int4.cpp)
  add_dependencies(example_gemm_xdl example_gemm_xdl_int4)
endif(USE_BITINT_EXTENSION_INT4)

60
61
62
# 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)
if(result EQUAL 0)
63
  add_dependencies(example_gemm_xdl example_gemm_xdl_fp64)
64
endif()
65

66
67
add_example_executable(example_gemm_xdl_streamk gemm_xdl_streamk.cpp)

68

zjing14's avatar
zjing14 committed
69
70
add_example_executable(example_gemm_xdl_fp8 gemm_xdl_fp8.cpp)
if(result EQUAL 0)
71
    add_dependencies(example_gemm_xdl example_gemm_xdl_fp8)
72
endif()
73

zjing14's avatar
zjing14 committed
74
75
add_example_executable(example_gemm_xdl_fp8_bf8 gemm_xdl_fp8_bf8.cpp)
if(result EQUAL 0)
76
    add_dependencies(example_gemm_xdl example_gemm_xdl_fp8_bf8)
77
endif()
78

79
add_example_executable(example_gemm_xdl_fp16_fp8 gemm_xdl_fp16_fp8.cpp)
80
if(result EQUAL 0)
zjing14's avatar
zjing14 committed
81
    add_dependencies(example_gemm_xdl example_gemm_xdl_fp16_fp8)
82
endif()
83
84


Jing Zhang's avatar
Jing Zhang committed
85
86
87
88
add_example_executable(example_gemm_xdl_input_i32_comp_i8_scale_ab gemm_xdl_input_i32_comp_i8_scale_ab.cpp)
if(result EQUAL 0)
    add_dependencies(example_gemm_xdl example_gemm_xdl_input_i32_comp_i8_scale_ab)
endif()