"vscode:/vscode.git/clone" did not exist on "dd900bf22c808a514e829837891d256e7bb8f401"
CMakeLists.txt 2.88 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
69
70
71

if(GPU_TARGETS MATCHES "gfx940" OR GPU_TARGETS MATCHES "gfx941" OR GPU_TARGETS MATCHES "gfx942")
  add_example_executable(example_gemm_xdl_f8 gemm_xdl_f8.cpp)
  if(result EQUAL 0)
72
73
    add_dependencies(example_gemm_xdl example_gemm_xdl_f8)
  endif()
74
endif()
75

76
77
add_example_executable(example_gemm_xdl_fp16_f8 gemm_xdl_fp16_f8.cpp)
if(result EQUAL 0)
78
79
  add_dependencies(example_gemm_xdl example_gemm_xdl_fp16_f8)
endif()