Skip to content

D
dlib
Commit 4a79d07b authored by Daniel Crispell's avatar Daniel Crispell
Browse files
Options

try_compile needs cudnn cmake vars explicitly passed

parent d207348a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
dlib/CMakeLists.txt
View file @ 4a79d07b
...@@ -467,7 +467,8 @@ if (NOT TARGET dlib) ...@@ -467,7 +467,8 @@ if (NOT TARGET dlib)
else() else()
message(STATUS "Checking if you have the right version of cuDNN installed.") message(STATUS "Checking if you have the right version of cuDNN installed.")
try_compile(cudnn_test_compile_worked ${PROJECT_BINARY_DIR}/cudnn_test_build try_compile(cudnn_test_compile_worked ${PROJECT_BINARY_DIR}/cudnn_test_build
${PROJECT_SOURCE_DIR}/dnn/test_for_cudnn cudnn_test) ${PROJECT_SOURCE_DIR}/dnn/test_for_cudnn cudnn_test
CMAKE_FLAGS -Dcudnn=${cudnn} -Dcudnn_include=${cudnn_include})
if (cudnn_test_compile_worked) if (cudnn_test_compile_worked)
message(STATUS "Found cuDNN: " ${cudnn}) message(STATUS "Found cuDNN: " ${cudnn})
else() else()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment