From 52dcade4aa38f3a2a04aa8f34d6535327a9bf260 Mon Sep 17 00:00:00 2001
From: Baizhou Zhang <sobereddiezhang@gmail.com>
Date: Wed, 4 Mar 2026 00:09:58 -0800
Subject: [PATCH] Fix flashinfer bump workflow (#19855)

---
 scripts/release/bump_flashinfer_version.py | 31 +++++++++++++++-------
 1 file changed, 22 insertions(+), 9 deletions(-)

diff --git a/scripts/release/bump_flashinfer_version.py b/scripts/release/bump_flashinfer_version.py
index 7f47374e5..27e4116ef 100755
--- a/scripts/release/bump_flashinfer_version.py
+++ b/scripts/release/bump_flashinfer_version.py
@@ -12,7 +12,8 @@ FILES_TO_UPDATE = [
     Path("docker/Dockerfile"),
     Path("scripts/ci/cuda/ci_install_dependency.sh"),
     Path("python/sglang/srt/entrypoints/engine.py"),
-    Path("python/sglang/srt/server_args.py"),
+    Path(".github/workflows/release-docker-cu13-framework.yml"),
+    Path("python/sglang/srt/utils/common.py"),
 ]
 
 
@@ -20,7 +21,9 @@ def read_current_flashinfer_version(repo_root: Path) -> str:
     """Read the current flashinfer version from python/pyproject.toml."""
     pyproject = repo_root / "python" / "pyproject.toml"
     content = pyproject.read_text()
-    match = re.search(r"flashinfer_python==(\d+\.\d+\.\d+)", content)
+    match = re.search(
+        r"flashinfer_python==(\d+\.\d+\.\d+(?:rc\d+|\.post\d+)?)", content
+    )
     if not match:
         raise ValueError(f"Could not find flashinfer_python version in {pyproject}")
     return match.group(1)
@@ -56,18 +59,28 @@ def replace_flashinfer_version(
             rf"\g<1>{new_version}",
             new_content,
         )
-    elif name == "engine.py" or name == "server_args.py":
+    elif name == "engine.py":
+        new_content = re.sub(
+            r'(assert_pkg_version\(\s*"flashinfer_python",\s*)"'
+            + re.escape(old_version)
+            + r'"',
+            r'\g<1>"' + new_version + '"',
+            new_content,
+            flags=re.DOTALL,
+        )
+    elif name == "release-docker-cu13-framework.yml":
         new_content = new_content.replace(
-            f'flashinfer-python", "{old_version}"',
-            f'flashinfer-python", "{new_version}"',
+            f"FlashInfer version (default: {old_version})",
+            f"FlashInfer version (default: {new_version})",
         )
         new_content = new_content.replace(
-            f'flashinfer_python", "{old_version}"',
-            f'flashinfer_python", "{new_version}"',
+            f'default: "{old_version}"',
+            f'default: "{new_version}"',
         )
+    elif name == "common.py":
         new_content = new_content.replace(
-            f"flashinfer-python version is >= {old_version}",
-            f"flashinfer-python version is >= {new_version}",
+            f'e.g., "{old_version}"',
+            f'e.g., "{new_version}"',
         )
 
     if content == new_content: