CMakeLists.txt 1000 Bytes
Newer Older
1
add_custom_target(example_batched_gemm_xdl)
2
3
add_example_executable(example_batched_gemm_xdl_fp32 batched_gemm_xdl_fp32.cpp)
if(result EQUAL 0)
4
5
  add_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_fp32)
endif()
6
7
add_example_executable(example_batched_gemm_xdl_fp16 batched_gemm_xdl_fp16.cpp)
if(result EQUAL 0)
8
9
  add_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_fp16)
endif()
10
11
12
add_example_executable(example_batched_gemm_xdl_bf16 batched_gemm_xdl_bf16.cpp)
if(result EQUAL 0)
  add_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_bf16)
13
endif()
14
15
add_example_executable(example_batched_gemm_xdl_int8 batched_gemm_xdl_int8.cpp)
if(result EQUAL 0)
16
17
  add_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_int8)
endif()
18
19
if(USE_BITINT_EXTENSION_INT4)
  add_example_executable(example_batched_gemm_xdl_int4 batched_gemm_xdl_int4.cpp)
20
21
22
  if(result EQUAL 0)
    add_dependencies(example_batched_gemm_xdl example_batched_gemm_xdl_int4)
  endif()
23
endif()