From f50cf07c7497484dcea1c022d47b4203b5f88dcb Mon Sep 17 00:00:00 2001 From: tzhouam Date: Fri, 13 Feb 2026 07:22:07 +0000 Subject: [PATCH] Update Dockerfile.ci to allow the installation of 'pplx_kernels' to continue even if the compatible wheel is not found, enhancing build resilience. Signed-off-by: tzhouam --- docker/Dockerfile.ci | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index bb4be84b5..560661761 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -26,7 +26,10 @@ RUN uv pip uninstall --system nvidia-cublas-cu12 # Use bash -c so that $(python3 -c ...) is expanded inside the container. RUN uv pip install --system --no-cache-dir ".[dev]" -RUN pip install --upgrade --force-reinstall "pplx_kernels" +# pplx_kernels is optional; the base image may already ship a version. +# Reinstall attempt may fail if no compatible wheel exists for the current +# PyTorch/CUDA combination – allow the build to continue regardless. +RUN pip install --upgrade --force-reinstall "pplx_kernels" || true RUN /bin/bash -c 'FLASHINFER_CUDA_TAG="$(python3 -c "import torch; print((torch.version.cuda or \"12.4\").replace(\".\", \"\"))")" && \ uv pip install --system --upgrade --force-reinstall \