File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -334,7 +334,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
334
334
uv pip install --system /wheels/xformers/*.whl --verbose
335
335
336
336
# 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'
338
338
# install package for build flashinfer
339
339
# see issue: https://github.com/flashinfer-ai/flashinfer/issues/738
340
340
@@ -353,7 +353,6 @@ RUN --mount=type=cache,target=/root/.cache/ccache \
353
353
git clone --depth 1 --recursive --shallow-submodules \
354
354
--branch ${FLASHINFER_GIT_REF} \
355
355
${FLASHINFER_GIT_REPO} flashinfer \
356
- && export UV_TORCH_BACKEND=cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.') \
357
356
&& echo "Building FlashInfer with AOT for arches FLASHINFER_CUDA_ARCH_LIST: ${FLASHINFER_CUDA_ARCH_LIST}, TORCH_CUDA_ARCH_LIST ${TORCH_CUDA_ARCH_LIST}" \
358
357
&& cd flashinfer \
359
358
&& python3 -m flashinfer.aot \
You can’t perform that action at this time.
0 commit comments