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