CMakeLists.txt 381 字节
Newer Older
1 2
set(DEV_LIBS custom_device)

3
# GPU
4
if(WITH_GPU OR WITH_ROCM)
5
  add_subdirectory(gpu)
6
endif()
7

8
# XPU
9
if(WITH_XPU)
10
  add_subdirectory(xpu)
11
endif()
12 13

# NPU
14
if(WITH_ASCEND OR WITH_ASCEND_CL)
15
  add_subdirectory(npu)
16
endif()
J
jianghaicheng 已提交
17 18

# IPU
19
if(WITH_IPU)
J
jianghaicheng 已提交
20
  add_subdirectory(ipu)
21
endif()
F
fwenguang 已提交
22 23

# MLU
24
if(WITH_MLU)
F
fwenguang 已提交
25
  add_subdirectory(mlu)
26
endif()
27 28 29 30

if(WITH_CUSTOM_DEVICE)
  add_subdirectory(custom)
endif()