Skip to content

GitLab

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

[Bugfix] Build hadacore kernels on >SM90 (#28748) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent f05d474c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
CMakeLists.txt
View file @ edfe4981
......@@ -861,7 +861,7 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
endif()
# Hadacore kernels
cuda_archs_loose_intersection(HADACORE_ARCHS "8.0;8.9;9.0" "${CUDA_ARCHS}")
cuda_archs_loose_intersection(HADACORE_ARCHS "8.0+PTX;9.0+PTX" "${CUDA_ARCHS}")
if(HADACORE_ARCHS)
set(SRCS "csrc/quantization/hadamard/hadacore/hadamard_transform_cuda.cu")
set_gencode_flags_for_srcs(
......
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