Skip to content

GitLab

Unverified Commit 7310aede authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix: compatible with PEP 440 (#2435)

parent 5de9a58e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
sgl-kernel/setup.py
View file @ 7310aede
......@@ -31,13 +31,13 @@ def rename_wheel():
zip_ref.extractall(tmp_dir)
old_info = tmp_dir / f"sgl_kernel-{base_version}.dist-info"
new_info = tmp_dir / f"sgl_kernel-{base_version}+cu{cuda_version}.dist-info"
new_info = tmp_dir / f"sgl_kernel-{base_version}.post0+cu{cuda_version}.dist-info"
old_info.rename(new_info)
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}"
base_version, f"{base_version}.post0+cu{cuda_version}"
)
with zipfile.ZipFile(new_wheel, "w", zipfile.ZIP_DEFLATED) as new_zip:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment