|
|
|
@ -137,9 +137,9 @@ set(EXTERNAL_LIBS
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
list(APPEND EXTERNAL_LIB ${CUDA_LIBRARIES} ${CUDA_rt_LIBRARY})
|
|
|
|
|
list(APPEND EXTERNAL_LIBS ${CUDA_LIBRARIES} ${CUDA_rt_LIBRARY})
|
|
|
|
|
if(NOT WITH_DSO)
|
|
|
|
|
list(APPEND EXTERNAL_LIB ${CUDNN_LIBRARY} ${CUDA_CUBLAS_LIBRARIES} ${CUDA_curand_LIBRARY})
|
|
|
|
|
list(APPEND EXTERNAL_LIBS ${CUDNN_LIBRARY} ${CUDA_CUBLAS_LIBRARIES} ${CUDA_curand_LIBRARY})
|
|
|
|
|
endif(NOT WITH_DSO)
|
|
|
|
|
endif(WITH_GPU)
|
|
|
|
|
|
|
|
|
|