CMakeLists.txt 581 Bytes
Newer Older
1
if(DTYPES MATCHES "int8" OR NOT DEFINED DTYPES)
2
# dlops
3
4
5
if(DL_KERNELS)
  add_example_executable(example_gemm_dl_quantization_int8 gemm_dl_quantization_int8.cpp)
endif()
6
7

# xdlops
8
9
10
11
list(APPEND gpu_list gfx908 gfx90a gfx940 gfx941 gfx942)
set(target 0)
foreach(gpu IN LISTS GPU_TARGETS)
 if(gpu IN_LIST gpu_list AND target EQUAL 0)
12
13
   add_example_executable(example_gemm_xdl_bias_relu_quantization_int8 gemm_xdl_bias_relu_quantization_int8.cpp)
   add_example_executable(example_gemm_xdl_quantization_int8 gemm_xdl_quantization_int8.cpp)
14
15
   set(target 1)
 endif()
16
17
endforeach()
endif()