[CUDA] Add arch=8.9 to CUDA_ARCHS for RTX 40XX (#6272)

ef2a49cd · Dmitry Ulyanov · GitHub · 2e3543c0 · ef2a49cd
Unverified Commit ef2a49cd authored Jan 12, 2024 by Dmitry Ulyanov Committed by GitHub Jan 12, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

CMakeLists.txt CMakeLists.txt +1 -0

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -215,6 +215,7 @@ if(USE_CUDA)
        list(APPEND CUDA_ARCHS "8.7")
    endif()
    if(CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
+        list(APPEND CUDA_ARCHS "8.9")
        list(APPEND CUDA_ARCHS "9.0")
    endif()
    list(POP_BACK CUDA_ARCHS CUDA_LAST_SUPPORTED_ARCH)