Skip to content

GitLab

Unverified Commit bf57ccc5 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

Remove sm120 arch from sm100 cutlass kernel arch list (#19716) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent ffb2cd6b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
CMakeLists.txt
View file @ bf57ccc5
...@@ -420,9 +420,9 @@ if(VLLM_GPU_LANG STREQUAL "CUDA") ...@@ -420,9 +420,9 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
endif() endif()
endif() endif()
# The cutlass_scaled_mm kernels for Blackwell (c3x, i.e. CUTLASS 3.x) require # The cutlass_scaled_mm kernels for Blackwell SM100 (c3x, i.e. CUTLASS 3.x)
# CUDA 12.8 or later # require CUDA 12.8 or later
cuda_archs_loose_intersection(SCALED_MM_ARCHS "10.0a;10.1a;12.0a" "${CUDA_ARCHS}") 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) if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER 12.8 AND SCALED_MM_ARCHS)
set(SRCS set(SRCS
"csrc/quantization/cutlass_w8a8/scaled_mm_c3x_sm100.cu" "csrc/quantization/cutlass_w8a8/scaled_mm_c3x_sm100.cu"
......
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