Added a check for CUDA before enabling it.

This commit is contained in:
Richard Musil 2019-09-19 15:48:31 +02:00
parent 7165f8ee51
commit dfcd43bc49

View File

@ -237,8 +237,10 @@ endif ()
if (${CMAKE_VERSION} VERSION_LESS 3.15) if (${CMAKE_VERSION} VERSION_LESS 3.15)
find_package(CUDA 9.0) find_package(CUDA 9.0)
else () else ()
enable_language(CUDA OPTIONAL) include(CheckLanguage)
check_language(CUDA)
if (CMAKE_CUDA_COMPILER) if (CMAKE_CUDA_COMPILER)
enable_language(CUDA OPTIONAL)
set (CUDA_FOUND TRUE) set (CUDA_FOUND TRUE)
endif () endif ()
endif () endif ()