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

W
Wilber 已提交
3
add_subdirectory(cpu)
4

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

W
Wilber 已提交
9 10 11 12 13 14 15 16 17
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 已提交
18 19 20 21

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