EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

fix: update update_wheel_index for cu128 (#5300)

Browse Source
This commit is contained in:
Yineng Zhang
2025-04-11 09:31:03 -07:00
committed by GitHub
parent 4aa6bab0b0
commit 80aa8ca84e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/update_kernel_whl_index.py
View File

@@ -14,7 +14,7 @@ def update_wheel_index(cuda_version="118"):
for path in sorted(pathlib.Path("sgl-kernel/dist").glob("*.whl")): for path in sorted(pathlib.Path("sgl-kernel/dist").glob("*.whl")):
with open(path, "rb") as f: with open(path, "rb") as f:
sha256 = hashlib.sha256(f.read()).hexdigest() sha256 = hashlib.sha256(f.read()).hexdigest()
ver = re.findall(r"sgl_kernel-([0-9.]+(?:\.post[0-9]+)?)-", path.name)[0] ver = re.findall(r"sgl_kernel-([0-9.]+(?:\.post[0-9]+)?)(?:\+cu[0-9]+)?-", path.name)[0]
full_url = f"{base_url}/v{ver}/{path.name}#sha256={sha256}" full_url = f"{base_url}/v{ver}/{path.name}#sha256={sha256}"
with (index_dir / "index.html").open("a") as f: with (index_dir / "index.html").open("a") as f:
f.write(f'<a href="{full_url}">{path.name}</a><br>\n') f.write(f'<a href="{full_url}">{path.name}</a><br>\n')