CMakeLists.txt 277 字节
Newer Older
Y
Yan Chunwei 已提交
1 2 3
if(NOT LITE_WITH_CUDA)
    return()
endif()
4
get_property(cuda_static_deps GLOBAL PROPERTY CUDA_STATIC_MODULES)
Y
Yan Chunwei 已提交
5

S
sangoly 已提交
6 7
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 已提交
8
 
9
add_subdirectory(math)