Skip to content

Commit 2b9022e

Browse files
committed
bump flashinfer to 0.5.0.rc1
Signed-off-by: jiahanc <[email protected]>
1 parent b98cbdb commit 2b9022e

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

docker/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -371,8 +371,8 @@ BASH
371371
# Install FlashInfer pre-compiled kernel cache and binaries
372372
# https://docs.flashinfer.ai/installation.html
373373
RUN --mount=type=cache,target=/root/.cache/uv \
374-
uv pip install --system flashinfer-cubin==0.4.1 \
375-
&& uv pip install --system flashinfer-jit-cache==0.4.1 \
374+
uv pip install --system flashinfer-cubin==0.5.0rc1 \
375+
&& uv pip install --system flashinfer-jit-cache==0.5.0rc1 \
376376
--extra-index-url https://flashinfer.ai/whl/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.') \
377377
&& flashinfer show-config
378378

requirements/cuda.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,4 @@ torchvision==0.24.0 # Required for phi3v processor. See https://github.com/pytor
1212
# https://github.com/facebookresearch/xformers/releases/tag/v0.0.32.post1
1313
# xformers==0.0.32.post1; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch >= 2.8
1414
# FlashInfer should be updated together with the Dockerfile
15-
flashinfer-python==0.4.1
15+
flashinfer-python==0.5.0rc1

0 commit comments

Comments
 (0)