add missing flag to CUDA_NVCC_FLAGS

This commit is contained in:
Marcin Junczys-Dowmunt 2018-03-04 17:50:21 -08:00
parent ea2a0aca65
commit 83e9e8927a

View File

@ -46,6 +46,7 @@ if(CUDA_FOUND)
endif(CUDNN_FOUND)
endif(USE_CUDNN)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DCUDA_FOUND")
list(APPEND CUDA_NVCC_FLAGS -DCUDA_FOUND; )
else(CUDA_FOUND)
message(WARNING "Cannot find CUDA compiling CPU version only")
endif(CUDA_FOUND)