We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5aecd78 commit e6e0733Copy full SHA for e6e0733
CMakeLists.txt
@@ -22,7 +22,10 @@ set(ignoreMe "${VLLM_PYTHON_PATH}")
22
set(PYTHON_SUPPORTED_VERSIONS "3.8" "3.9" "3.10" "3.11" "3.12")
23
24
# Supported NVIDIA architectures.
25
-set(CUDA_SUPPORTED_ARCHS "8.0;8.6;8.9;9.0;10.0;10.1;12.0")
+set(CUDA_SUPPORTED_ARCHS "8.0;8.6;8.9;9.0")
26
+if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER_EQUAL 12.8)
27
+ list(APPEND CUDA_SUPPORTED_ARCHS "10.0" "10.1" "12.0")
28
+endif()
29
30
# Supported AMD GPU architectures.
31
set(HIP_SUPPORTED_ARCHS "gfx906;gfx908;gfx90a;gfx940;gfx941;gfx942;gfx1030;gfx1100")
0 commit comments