CMakeLists.txt 450 字节
Newer Older
Q
qijun 已提交
1 2
add_subdirectory(dynload)

Y
Yu Yang 已提交
3
nv_test(cuda_test SRCS cuda_test.cu DEPS dyload_cuda)
4

F
fengjiayi 已提交
5
cc_library(place SRCS place.cc)
L
liaogang 已提交
6
cc_test(place_test SRCS place_test.cc DEPS place glog gflags)
Y
Yu Yang 已提交
7 8 9 10 11
IF(WITH_GPU)
    set(GPU_CTX_DEPS dyload_cuda dynamic_loader )
ELSE()
    set(GPU_CTX_DEPS)
ENDIF()
Q
QI JUN 已提交
12

Y
Yu Yang 已提交
13 14
cc_library(device_context SRCS device_context.cc DEPS place eigen3 ${GPU_CTX_DEPS})
nv_test(device_context_test SRCS device_context_test.cc DEPS device_context glog gflags)