Skip to content

Commit e6e0733

Browse files
committed
guarding blackwell build against cuda < 12.8
1 parent 5aecd78 commit e6e0733

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

CMakeLists.txt

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,10 @@ set(ignoreMe "${VLLM_PYTHON_PATH}")
2222
set(PYTHON_SUPPORTED_VERSIONS "3.8" "3.9" "3.10" "3.11" "3.12")
2323

2424
# Supported NVIDIA architectures.
25-
set(CUDA_SUPPORTED_ARCHS "8.0;8.6;8.9;9.0;10.0;10.1;12.0")
25+
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()
2629

2730
# Supported AMD GPU architectures.
2831
set(HIP_SUPPORTED_ARCHS "gfx906;gfx908;gfx90a;gfx940;gfx941;gfx942;gfx1030;gfx1100")

0 commit comments

Comments
 (0)