From 4d8a76e7214c6c7f036e3039050a44ab206f0730 Mon Sep 17 00:00:00 2001 From: tzhouam Date: Tue, 3 Mar 2026 09:23:09 +0000 Subject: [PATCH] Reintroduce installation of flashinfer-jit-cache in Dockerfile.ci to enhance dependency management for the CI environment, ensuring compatibility with the latest requirements. Signed-off-by: tzhouam --- docker/Dockerfile.ci | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index 444cdbfc1..2f62102bf 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -25,6 +25,9 @@ RUN uv pip install --system --upgrade \ "flashinfer-cubin==0.6.4" \ "nvidia-cublas-cu12==12.9.1.4" \ "numpy==2.2.6" +RUN uv pip install --system --upgrade \ + "flashinfer-jit-cache==0.6.4" \ + --index-url https://flashinfer.ai/whl/cu129 RUN ln -sf /usr/bin/python3 /usr/bin/python ENTRYPOINT []