ci: refactor CUDA dependency install script (#21017)

This commit is contained in:
Lianmin Zheng
2026-03-20 21:36:58 -07:00
committed by GitHub
parent d5473f87c4
commit 50404e0d1f
5 changed files with 325 additions and 186 deletions
@@ -10,7 +10,6 @@ from utils import compare_versions, get_repo_root, normalize_version, validate_v
FILES_TO_UPDATE = [
Path("python/pyproject.toml"),
Path("docker/Dockerfile"),
Path("scripts/ci/cuda/ci_install_dependency.sh"),
Path("python/sglang/srt/entrypoints/engine.py"),
Path("python/sglang/srt/utils/common.py"),
]
@@ -52,12 +51,6 @@ def replace_flashinfer_version(
rf"\g<1>{new_version}",
new_content,
)
elif name == "ci_install_dependency.sh":
new_content = re.sub(
rf"(FLASHINFER_VERSION=){re.escape(old_version)}",
rf"\g<1>{new_version}",
new_content,
)
elif name == "engine.py":
new_content = re.sub(
r'(assert_pkg_version\(\s*"flashinfer_python",\s*)"'