From 80aa8ca84eb4391bc14e4fa2f4e85999b5c1e14f Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Fri, 11 Apr 2025 09:31:03 -0700 Subject: [PATCH] fix: update update_wheel_index for cu128 (#5300) --- scripts/update_kernel_whl_index.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/update_kernel_whl_index.py b/scripts/update_kernel_whl_index.py index 8e0a2b5f3..823a58f5d 100644 --- a/scripts/update_kernel_whl_index.py +++ b/scripts/update_kernel_whl_index.py @@ -14,7 +14,7 @@ def update_wheel_index(cuda_version="118"): for path in sorted(pathlib.Path("sgl-kernel/dist").glob("*.whl")): with open(path, "rb") as f: 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}" with (index_dir / "index.html").open("a") as f: f.write(f'{path.name}
\n')