Skip to content

GitLab

Unverified Commit f3f6545d authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

[ci] add support for 8.0 and 8.6 CUDA archs (#4454)

parent d05f5470
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
CMakeLists.txt
View file @ f3f6545d
...@@ -174,6 +174,12 @@ if(USE_CUDA) ...@@ -174,6 +174,12 @@ if(USE_CUDA)
if(CUDA_VERSION VERSION_GREATER_EQUAL "10.0") if(CUDA_VERSION VERSION_GREATER_EQUAL "10.0")
list(APPEND CUDA_ARCHS "7.5") list(APPEND CUDA_ARCHS "7.5")
endif() endif()
if(CUDA_VERSION VERSION_GREATER_EQUAL "11.0")
list(APPEND CUDA_ARCHS "8.0")
endif()
if(CUDA_VERSION VERSION_GREATER_EQUAL "11.1")
list(APPEND CUDA_ARCHS "8.6")
endif()
list(POP_BACK CUDA_ARCHS CUDA_LAST_SUPPORTED_ARCH) list(POP_BACK CUDA_ARCHS CUDA_LAST_SUPPORTED_ARCH)
list(APPEND CUDA_ARCHS "${CUDA_LAST_SUPPORTED_ARCH}+PTX") list(APPEND CUDA_ARCHS "${CUDA_LAST_SUPPORTED_ARCH}+PTX")
CUDA_SELECT_NVCC_ARCH_FLAGS(CUDA_ARCH_FLAGS ${CUDA_ARCHS}) CUDA_SELECT_NVCC_ARCH_FLAGS(CUDA_ARCH_FLAGS ${CUDA_ARCHS})
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment