From 50cea0b29caada47e8a7f6abbacb291bc5e7604f Mon Sep 17 00:00:00 2001 From: tzhouam Date: Wed, 11 Feb 2026 04:34:52 +0000 Subject: [PATCH] Refactor flashinfer version retrieval in Dockerfile.ci for improved accuracy and consistency in package installations Signed-off-by: tzhouam --- docker/Dockerfile.ci | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index b6df5c223..c2c60267f 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -21,18 +21,9 @@ RUN apt-get update && \ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm --extra-index-url https://wheels.vllm.ai/4a1550d22d7058e129d0e1257e726b3bf4a77025 # Keep flashinfer runtime packages in version sync. -# We derive the version from installed flashinfer-python (pulled by vllm) +# We derive the version from installed flashinfer (pulled by vllm) # and enforce the same version for flashinfer-cubin / flashinfer-jit-cache. -RUN FLASHINFER_VERSION="$(python3 -c 'import importlib.metadata as m; \ -for n in ("flashinfer", "flashinfer-python"): \ - \ - try: \ - print(m.version(n)); \ - break; \ - except m.PackageNotFoundError: \ - pass \ -else: \ - raise SystemExit("flashinfer package not found")')" && \ +RUN FLASHINFER_VERSION="$(python3 -c 'import importlib.metadata as m; print(next(m.version(n) for n in ("flashinfer","flashinfer-python") if m.packages_distributions().get(n)))')" && \ FLASHINFER_CUDA_TAG="$(python3 -c 'import torch; print((torch.version.cuda or "12.4").replace(".", ""))')" && \ uv pip install --system --upgrade --force-reinstall \ "flashinfer-cubin==${FLASHINFER_VERSION}" \