Skip to content

Commit 4220e2a

Browse files
mgoingeodavic
authored andcommitted
[Bugfix] Build hadacore kernels on >SM90 (vllm-project#28748)
Signed-off-by: mgoin <mgoin64@gmail.com> Signed-off-by: George D. Torres <gdavtor@gmail.com>
1 parent d11923a commit 4220e2a

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
@@ -861,7 +861,7 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
861861
endif()
862862

863863
# Hadacore kernels
864-
cuda_archs_loose_intersection(HADACORE_ARCHS "8.0;8.9;9.0" "${CUDA_ARCHS}")
864+
cuda_archs_loose_intersection(HADACORE_ARCHS "8.0+PTX;9.0+PTX" "${CUDA_ARCHS}")
865865
if(HADACORE_ARCHS)
866866
set(SRCS "csrc/quantization/hadamard/hadacore/hadamard_transform_cuda.cu")
867867
set_gencode_flags_for_srcs(

0 commit comments

Comments
 (0)