Skip to content

GitLab

Unverified Commit ef2a49cd authored by Dmitry Ulyanov's avatar Dmitry Ulyanov Committed by GitHub
Browse files

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

parent 2e3543c0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
CMakeLists.txt
View file @ ef2a49cd
...@@ -215,6 +215,7 @@ if(USE_CUDA) ...@@ -215,6 +215,7 @@ if(USE_CUDA)
list(APPEND CUDA_ARCHS "8.7") list(APPEND CUDA_ARCHS "8.7")
endif() endif()
if(CUDA_VERSION VERSION_GREATER_EQUAL "11.8") if(CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
list(APPEND CUDA_ARCHS "8.9")
list(APPEND CUDA_ARCHS "9.0") list(APPEND CUDA_ARCHS "9.0")
endif() endif()
list(POP_BACK CUDA_ARCHS CUDA_LAST_SUPPORTED_ARCH) list(POP_BACK CUDA_ARCHS CUDA_LAST_SUPPORTED_ARCH)
......
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