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

f3f6545d · Nikita Titov · GitHub · d05f5470 · f3f6545d
Unverified Commit f3f6545d authored Jul 10, 2021 by Nikita Titov Committed by GitHub Jul 10, 2021
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

CMakeLists.txt CMakeLists.txt +6 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -174,6 +174,12 @@ if(USE_CUDA)
    if(CUDA_VERSION VERSION_GREATER_EQUAL "10.0")
        list(APPEND CUDA_ARCHS "7.5")
    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(APPEND CUDA_ARCHS "${CUDA_LAST_SUPPORTED_ARCH}+PTX")
    CUDA_SELECT_NVCC_ARCH_FLAGS(CUDA_ARCH_FLAGS ${CUDA_ARCHS})