Skip to content

Commit 700d608

Browse files
authored
Remove somewhat unnecessary change
1 parent 1b27857 commit 700d608

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

.github/ci_configs/vllm/Dockerfile.tmp_vllm

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
334334
uv pip install --system /wheels/xformers/*.whl --verbose
335335

336336
# Build flashinfer from source.
337-
ARG torch_cuda_arch_list="8.0;8.9;9.0a;10.0a;12.0"
337+
ARG torch_cuda_arch_list='8.0;8.9;9.0a;10.0a;12.0'
338338
# install package for build flashinfer
339339
# see issue: https://github.com/flashinfer-ai/flashinfer/issues/738
340340

@@ -353,7 +353,6 @@ RUN --mount=type=cache,target=/root/.cache/ccache \
353353
git clone --depth 1 --recursive --shallow-submodules \
354354
--branch ${FLASHINFER_GIT_REF} \
355355
${FLASHINFER_GIT_REPO} flashinfer \
356-
&& export UV_TORCH_BACKEND=cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.') \
357356
&& echo "Building FlashInfer with AOT for arches FLASHINFER_CUDA_ARCH_LIST: ${FLASHINFER_CUDA_ARCH_LIST}, TORCH_CUDA_ARCH_LIST ${TORCH_CUDA_ARCH_LIST}" \
358357
&& cd flashinfer \
359358
&& python3 -m flashinfer.aot \

0 commit comments

Comments
 (0)