diff options
| -rw-r--r-- | CMakeLists.txt | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 015f7c2..9ae7b26 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -79,6 +79,10 @@ if(CUDA_ENABLE) if(NOT CUDA_VERSION VERSION_LESS 8.0) list(APPEND DEFAULT_CUDA_ARCH "60" "61" "62") endif() + # add Volta support for CUDA >= 9.0 + if(NOT CUDA_VERSION VERSION_LESS 9.0) + list(APPEND DEFAULT_CUDA_ARCH "70") + endif() set(CUDA_ARCH "${DEFAULT_CUDA_ARCH}" CACHE STRING "Set GPU architecture (semicolon separated list, e.g. '-DCUDA_ARCH=20;35;60')") # validate architectures (only numbers are allowed) |
