Skip to content

GitLab

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

[CI Bugfix] Specify same TORCH_CUDA_ARCH_LIST for flashinfer aot and install (#20772) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent b854321f
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
docker/Dockerfile
View file @ cf75cd20
......@@ -408,6 +408,7 @@ RUN --mount=type=cache,target=/root/.cache/uv bash - <<'BASH'
# Needed to build AOT kernels
pushd flashinfer
TORCH_CUDA_ARCH_LIST="${FI_TORCH_CUDA_ARCH_LIST}" \
python3 -m flashinfer.aot
TORCH_CUDA_ARCH_LIST="${FI_TORCH_CUDA_ARCH_LIST}" \
uv pip install --system --no-build-isolation .
......
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