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
Hide 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") ...@@ -861,7 +861,7 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
endif() endif()
# Hadacore kernels # 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) if(HADACORE_ARCHS)
set(SRCS "csrc/quantization/hadamard/hadacore/hadamard_transform_cuda.cu") set(SRCS "csrc/quantization/hadamard/hadacore/hadamard_transform_cuda.cu")
set_gencode_flags_for_srcs( 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