Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dlib
Commits
6aa96fde
Commit
6aa96fde
authored
Dec 19, 2016
by
Juha Reunanen
Browse files
Set CUDA_HOST_COMPILER when not targeting Visual Studio
parent
33e58893
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
8 deletions
+11
-8
dlib/CMakeLists.txt
dlib/CMakeLists.txt
+11
-8
No files found.
dlib/CMakeLists.txt
View file @
6aa96fde
...
...
@@ -526,12 +526,18 @@ if (NOT TARGET dlib)
if
(
cudnn AND NOT DEFINED cuda_test_compile_worked AND NOT DEFINED cudnn_test_compile_worked
)
# make sure cuda is really working by doing a test compile
message
(
STATUS
"Building a CUDA test project to see if your compiler is compatible with CUDA..."
)
set
(
CUDA_TEST_CMAKE_FLAGS
"-DCMAKE_PREFIX_PATH=
${
CMAKE_PREFIX_PATH
}
"
"-DCMAKE_INCLUDE_PATH=
${
CMAKE_INCLUDE_PATH
}
"
"-DCMAKE_LIBRARY_PATH=
${
CMAKE_LIBRARY_PATH
}
"
)
if
(
!MSVC
)
# see https://github.com/davisking/dlib/issues/363
list
(
APPEND CUDA_TEST_CMAKE_FLAGS
"-DCUDA_HOST_COMPILER=
${
CUDA_HOST_COMPILER
}
"
)
endif
()
try_compile
(
cuda_test_compile_worked
${
PROJECT_BINARY_DIR
}
/cuda_test_build
${
PROJECT_SOURCE_DIR
}
/cmake_utils/test_for_cuda cuda_test
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}"
CMAKE_FLAGS
${
CUDA_TEST_CMAKE_FLAGS
}
)
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. ***"
)
...
...
@@ -539,10 +545,7 @@ if (NOT TARGET dlib)
message
(
STATUS
"Checking if you have the right version of cuDNN installed."
)
try_compile
(
cudnn_test_compile_worked
${
PROJECT_BINARY_DIR
}
/cudnn_test_build
${
PROJECT_SOURCE_DIR
}
/cmake_utils/test_for_cudnn cudnn_test
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
}
"
CMAKE_FLAGS
${
CUDA_TEST_CMAKE_FLAGS
}
)
if
(
cudnn_test_compile_worked
)
message
(
STATUS
"Found cuDNN: "
${
cudnn
}
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment