CMakeLists.txt 603 字节
Newer Older
L
liaogang 已提交
1 2
cc_library(cpu_info SRCS cpu_info.cc DEPS gflags glog)
cc_test(cpu_info_test SRCS cpu_info_test.cc DEPS cpu_info)
L
liaogang 已提交
3

L
liaogang 已提交
4
nv_library(gpu_info SRCS gpu_info.cc DEPS gflags)
5

F
fengjiayi 已提交
6
cc_library(place SRCS place.cc)
L
liaogang 已提交
7
cc_test(place_test SRCS place_test.cc DEPS place glog gflags)
L
liaogang 已提交
8

L
liaogang 已提交
9
add_subdirectory(dynload)
L
liaogang 已提交
10

L
liaogang 已提交
11 12
cc_test(enforce_test SRCS enforce_test.cc)

Q
qijun 已提交
13 14 15 16 17 18 19
IF(WITH_GPU)
    set(GPU_CTX_DEPS dynload_cuda dynamic_loader)
ELSE()
    set(GPU_CTX_DEPS)
ENDIF()

cc_library(device_context SRCS device_context.cc DEPS place eigen3 ${GPU_CTX_DEPS})
L
liaogang 已提交
20
nv_test(device_context_test SRCS device_context_test.cc DEPS device_context gpu_info)