Skip to content
Snippets Groups Projects
Commit d7209b21 authored by psychocrypt's avatar psychocrypt
Browse files

add NVIDIA Volta support

add `sm_70` if CUDA 9+ is avalable
parent 6701b0c2
No related branches found
No related tags found
Loading
......@@ -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)
......
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