CMakeLists.txt 306 字节
Newer Older
Y
Yan Chunwei 已提交
1 2 3
if(NOT LITE_WITH_CUDA)
    return()
endif()
S
sangoly 已提交
4 5
set(cuda_static_deps cudnn_static cublas_static curand_static
    culibos_static cudart_static)
Y
Yan Chunwei 已提交
6

S
sangoly 已提交
7 8
nv_library(target_wrapper_cuda SRCS target_wrapper.cc DEPS ${cuda_static_deps})
nv_library(cuda_blas SRCS blas.cc DEPS ${cuda_static_deps})
Y
Yan Chunwei 已提交
9
 
10
add_subdirectory(math)