Skip to content

GitLab

Commit a8c51cea authored by hannometer's avatar hannometer Committed by Davis E. King
Browse files

pass CUDA_HOST_COMPILER variable to cuda cmake tests (#292)

parent 41bc867a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
dlib/CMakeLists.txt
View file @ a8c51cea
......@@ -481,6 +481,7 @@ if (NOT TARGET dlib)
CMAKE_FLAGS "-DCMAKE_PREFIX_PATH=${CMAKE_PREFIX_PATH}"
"-DCMAKE_INCLUDE_PATH=${CMAKE_INCLUDE_PATH}"
"-DCMAKE_LIBRARY_PATH=${CMAKE_LIBRARY_PATH}"
"-DCUDA_HOST_COMPILER=${CUDA_HOST_COMPILER}"
)
if (NOT cuda_test_compile_worked)
message(STATUS "*** CUDA was found but your compiler failed to compile a simple CUDA program so dlib isn't going to use CUDA. ***")
......@@ -491,6 +492,7 @@ if (NOT TARGET dlib)
CMAKE_FLAGS "-DCMAKE_PREFIX_PATH=${CMAKE_PREFIX_PATH}"
"-DCMAKE_INCLUDE_PATH=${CMAKE_INCLUDE_PATH}"
"-DCMAKE_LIBRARY_PATH=${CMAKE_LIBRARY_PATH}"
"-DCUDA_HOST_COMPILER=${CUDA_HOST_COMPILER}"
)
if (cudnn_test_compile_worked)
message(STATUS "Found cuDNN: " ${cudnn})
......
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