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

fix: use manylinux2014_x86_64 tag (#2434)

Browse Source
This commit is contained in:
Yineng Zhang
2024-12-11 06:17:41 +08:00
committed by GitHub
parent 56fcd8e8a5
commit 5de9a58eca
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
sgl-kernel/setup.py
View File

@@ -34,10 +34,12 @@ def rename_wheel():
new_info = tmp_dir / f"sgl_kernel-{base_version}+cu{cuda_version}.dist-info"
old_info.rename(new_info)
new_wheel = (
wheel_dir
/ f"sgl_kernel-{base_version}+cu{cuda_version}-{old_wheel.name.split('-', 2)[-1]}"
platform = "manylinux2014_x86_64"
new_wheel = wheel_dir / old_wheel.name.replace("linux_x86_64", platform)
new_wheel = wheel_dir / new_wheel.name.replace(
base_version, f"{base_version}+cu{cuda_version}"
)
with zipfile.ZipFile(new_wheel, "w", zipfile.ZIP_DEFLATED) as new_zip:
for file_path in tmp_dir.rglob("*"):
if file_path.is_file():