saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-05 04:17:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix cuda build for cmake >= 3.17.0 (#3362)

Browse source
This commit is contained in:
George Wu 2020-03-30 15:38:57 +08:00 committed by GitHub
parent 33b5010e62
commit 355f39ddee
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cmake/CMakeLists.txt
View file

@ -711,7 +711,12 @@ if (onnxruntime_USE_CUDA)
if ( CMAKE_CUDA_FLAGS )
list(REMOVE_ITEM CMAKE_CUDA_FLAGS "-cudart static")
endif()
string(APPEND CMAKE_CUDA_FLAGS "-cudart shared")
if (${CMAKE_VERSION} VERSION_GREATER_EQUAL "3.17.0")
# cmake 3.17.0 introduces CMAKE_CUDA_RUNTIME which must be used instead of explicit -cudart shared flag
set(CMAKE_CUDA_RUNTIME_LIBRARY Shared)
else()
string(APPEND CMAKE_CUDA_FLAGS "-cudart shared")
endif()
enable_language(CUDA)
set(CMAKE_CUDA_STANDARD 11)
file(TO_CMAKE_PATH ${onnxruntime_CUDNN_HOME} onnxruntime_CUDNN_HOME)