Skip to content
Snippets Groups Projects
Commit da6ff1b9 authored by fireice-uk's avatar fireice-uk Committed by GitHub
Browse files

Merge pull request #59 from psychocrypt/topic-addVoltaSupport

add NVIDIA Volta support
parents bd31e5b0 d7209b21
No related branches found
No related tags found
No related merge requests found
...@@ -79,6 +79,10 @@ if(CUDA_ENABLE) ...@@ -79,6 +79,10 @@ if(CUDA_ENABLE)
if(NOT CUDA_VERSION VERSION_LESS 8.0) if(NOT CUDA_VERSION VERSION_LESS 8.0)
list(APPEND DEFAULT_CUDA_ARCH "60" "61" "62") list(APPEND DEFAULT_CUDA_ARCH "60" "61" "62")
endif() 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')") 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) # validate architectures (only numbers are allowed)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment