From bf57ccc5c2102a77d70431b2fe47e6fde4acc600 Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Wed, 18 Jun 2025 03:49:39 +0900 Subject: [PATCH] Remove sm120 arch from sm100 cutlass kernel arch list (#19716) Signed-off-by: mgoin --- CMakeLists.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index d75f0d321247..402131b7a1e7 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -420,9 +420,9 @@ if(VLLM_GPU_LANG STREQUAL "CUDA") endif() endif() - # The cutlass_scaled_mm kernels for Blackwell (c3x, i.e. CUTLASS 3.x) require - # CUDA 12.8 or later - cuda_archs_loose_intersection(SCALED_MM_ARCHS "10.0a;10.1a;12.0a" "${CUDA_ARCHS}") + # The cutlass_scaled_mm kernels for Blackwell SM100 (c3x, i.e. CUTLASS 3.x) + # require CUDA 12.8 or later + cuda_archs_loose_intersection(SCALED_MM_ARCHS "10.0a;10.1a" "${CUDA_ARCHS}") if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER 12.8 AND SCALED_MM_ARCHS) set(SRCS "csrc/quantization/cutlass_w8a8/scaled_mm_c3x_sm100.cu"