Skip to content

GitLab

Unverified Commit 205b9494 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[CI/Build] fix TORCH_CUDA_ARCH_LIST in wheel build (#3801)

parent 3bec41f4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
.github/workflows/scripts/build.sh
View file @ 205b9494
...@@ -15,6 +15,7 @@ $python_executable -m pip install -r requirements.txt ...@@ -15,6 +15,7 @@ $python_executable -m pip install -r requirements.txt
export MAX_JOBS=1 export MAX_JOBS=1
# Make sure punica is built for the release (for LoRA) # Make sure punica is built for the release (for LoRA)
export VLLM_INSTALL_PUNICA_KERNELS=1 export VLLM_INSTALL_PUNICA_KERNELS=1
# Make sure release wheels are built for the following architectures
export TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 8.6 8.9 9.0+PTX"
# Build # Build
$python_executable setup.py bdist_wheel --dist-dir=dist $python_executable setup.py bdist_wheel --dist-dir=dist
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