From 22a2bd3406f8004820b2b9c8849db699c565cc12 Mon Sep 17 00:00:00 2001 From: Zhou Taichang Date: Mon, 16 Mar 2026 19:31:03 +0800 Subject: [PATCH] Upgrade flashinfer-cubin and flashinfer-jit-cache versions Signed-off-by: Zhou Taichang --- docker/Dockerfile.ci | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index cb301f898..1397c8f90 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -21,12 +21,12 @@ RUN git clone https://github.com/vllm-project/vllm.git /workspace/vllm && \ cd ${APP_DIR} && \ uv pip install --system ".[dev]" && \ uv pip install --system --upgrade \ - "flashinfer-cubin==0.6.4" \ + "flashinfer-cubin==0.6.6" \ "nvidia-cublas-cu12==12.9.1.4" \ "numpy==2.2.6" RUN uv pip install --system --upgrade \ - "flashinfer-jit-cache==0.6.4" \ + "flashinfer-jit-cache==0.6.6" \ --index-url https://flashinfer.ai/whl/cu129 RUN ln -sf /usr/bin/python3 /usr/bin/python