diff --git a/docker/Dockerfile b/docker/Dockerfile index 055535f85..7063b2173 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -19,7 +19,7 @@ ARG PIP_DEFAULT_INDEX ARG UBUNTU_MIRROR ARG GITHUB_ARTIFACTORY=github.com ARG INSTALL_FLASHINFER_JIT_CACHE=0 -ARG FLASHINFER_VERSION=0.6.4 +ARG FLASHINFER_VERSION=0.6.6 ARG MOONCAKE_VERSION=0.3.9 #if need other arg please add in MOONCAKE_COMPILE_ARG ARG MOONCAKE_COMPILE_ARG="-DUSE_HTTP=ON -DUSE_MNNVL=ON -DUSE_CUDA=ON -DWITH_EP=ON" diff --git a/python/pyproject.toml b/python/pyproject.toml index cde1cab99..9b2613ddb 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -27,8 +27,8 @@ dependencies = [ "datasets", "einops", "fastapi", - "flashinfer_python==0.6.4", # keep it aligned with jit-cache version in Dockerfile - "flashinfer_cubin==0.6.4", + "flashinfer_python==0.6.6", # keep it aligned with jit-cache version in Dockerfile + "flashinfer_cubin==0.6.6", "gguf", "hf_transfer", "huggingface_hub", diff --git a/python/sglang/srt/entrypoints/engine.py b/python/sglang/srt/entrypoints/engine.py index dc1efbfc6..29ffc0da8 100644 --- a/python/sglang/srt/entrypoints/engine.py +++ b/python/sglang/srt/entrypoints/engine.py @@ -1135,7 +1135,7 @@ def _set_envs_and_config(server_args: ServerArgs): if server_args.attention_backend == "flashinfer": assert_pkg_version( "flashinfer_python", - "0.6.4", + "0.6.6", "Please uninstall the old version and " "reinstall the latest version by following the instructions " "at https://docs.flashinfer.ai/installation.html.", diff --git a/python/sglang/srt/utils/common.py b/python/sglang/srt/utils/common.py index 78fdb8a12..925552ede 100644 --- a/python/sglang/srt/utils/common.py +++ b/python/sglang/srt/utils/common.py @@ -1132,7 +1132,7 @@ def check_pkg_version_at_least(pkg: str, min_version: str) -> bool: Args: pkg: Package name (distribution name, e.g., "flashinfer-python") - min_version: Minimum version required (e.g., "0.6.4") + min_version: Minimum version required (e.g., "0.6.6") Returns: True if package is installed and version >= min_version, False otherwise diff --git a/scripts/ci/cuda/ci_install_dependency.sh b/scripts/ci/cuda/ci_install_dependency.sh index 7422bb765..381a05dc9 100755 --- a/scripts/ci/cuda/ci_install_dependency.sh +++ b/scripts/ci/cuda/ci_install_dependency.sh @@ -5,7 +5,7 @@ set -euxo pipefail # Set up environment variables IS_BLACKWELL=${IS_BLACKWELL:-0} CU_VERSION="cu129" -FLASHINFER_VERSION=0.6.4 +FLASHINFER_VERSION=0.6.6 OPTIONAL_DEPS="${1:-}" # Detect system architecture