8000 Update flashinfer build · pytorch/pytorch@754cd75 · GitHub
[go: up one dir, main page]
Skip to content

Commit 754cd75

Browse files
huydhnhuydhn
committed
Update flashinfer build
Signed-off-by: Huy Do <huydhn@gmail.com>
1 parent 2640092 commit 754cd75

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎.github/ci_configs/vllm/Dockerfile.tmp_vllm‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
341341
uv pip install --system /wheels/xformers/*.whl --verbose
342342

343343
# Build flashinfer from source.
344-
ARG torch_cuda_arch_list='8.0;8.9;9.0a'
344+
ARG torch_cuda_arch_list='8.0;8.9;9.0a;10.0a;12.0'
345345
# install package for build flashinfer
346346
# see issue: https://github.com/flashinfer-ai/flashinfer/issues/738
347347

0 commit comments

Comments
 (0)
0