CMakeLists.txt 323 字节
Newer Older
1 2 3

set(DEV_LIBS custom_device)

4 5 6 7 8
# GPU
IF(WITH_GPU OR WITH_ROCM)
  add_subdirectory(gpu)
ENDIF()

9 10 11 12
# XPU
IF(WITH_XPU)
  add_subdirectory(xpu)
ENDIF()
13 14 15 16 17

# NPU
IF(WITH_ASCEND OR WITH_ASCEND_CL)
  add_subdirectory(npu)
ENDIF()
J
jianghaicheng 已提交
18 19 20 21 22

# IPU
IF(WITH_IPU)
  add_subdirectory(ipu)
ENDIF()
F
fwenguang 已提交
23 24 25 26 27

# MLU
IF(WITH_MLU)
  add_subdirectory(mlu)
ENDIF()