Skip to content

Commit 533c217

Browse files
committed
Fix cutlass sm_90a vesrion in CMakeList
1 parent 6d21fa1 commit 533c217

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
200200
# The CUTLASS kernels for Hopper require sm90a to be enabled.
201201
# This is done via the below gencode option, BUT that creates kernels for both sm90 and sm90a.
202202
# That adds an extra 17MB to compiled binary, so instead we selectively enable it.
203-
if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER 11)
203+
if(${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER 12.0)
204204
set_source_files_properties(
205205
"csrc/quantization/cutlass_w8a8/scaled_mm_dq_c3x.cu"
206206
PROPERTIES

0 commit comments

Comments
 (0)