CMakeLists.txt 821 Bytes
Newer Older
1
2
add_custom_target(example_cgemm_xdl)

3
4
add_example_executable(example_cgemm_xdl_bf16 cgemm_xdl_bf16.cpp)
if(result EQUAL 0)
5
6
  add_dependencies(example_cgemm_xdl example_cgemm_xdl_bf16)
endif()
7
8
add_example_executable(example_cgemm_xdl_fp16 cgemm_xdl_fp16.cpp)
if(result EQUAL 0)
9
10
  add_dependencies(example_cgemm_xdl example_cgemm_xdl_fp16)
endif()
11
add_example_executable(example_cgemm_xdl_fp32 cgemm_xdl_fp32.cpp)
12
13
if(result EQUAL 0)
  add_dependencies(example_cgemm_xdl example_cgemm_xdl_fp32)
14
endif()
15
16
add_example_executable(example_cgemm_xdl_int8 cgemm_xdl_int8.cpp)
if(result EQUAL 0)
17
18
  add_dependencies(example_cgemm_xdl example_cgemm_xdl_int8)
endif()
19
20
21
22
if(USE_BITINT_EXTENSION_INT4)
  add_example_executable(example_cgemm_xdl_int4 cgemm_xdl_int4.cpp)
  add_dependencies(example_cgemm_xdl example_cgemm_xdl_int4)
endif()