CMakeLists.txt 392 字节
Newer Older
1 2
add_subdirectory(dynload)

W
Wilber 已提交
3
add_subdirectory(cpu)
4

5 6
add_subdirectory(custom)

W
Wilber 已提交
7 8 9 10
if(WITH_GPU OR WITH_ROCM)
  add_subdirectory(gpu)
endif()

W
Wilber 已提交
11 12 13 14 15 16 17 18 19
if(WITH_XPU)
  add_subdirectory(xpu)
endif()

cc_library(pten_context SRCS all_context.cc DEPS device_context cpu_context)

if(WITH_XPU)
  add_dependencies(pten_context xpu_context)
endif()
W
Wilber 已提交
20 21 22 23

if(WITH_GPU)
  add_dependencies(pten_context gpu_context)
endif()