Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "b3c437e009b1efc44c56f52e62c298ff9acd727c"
Unverified Commit 5de9a58e authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix: use manylinux2014_x86_64 tag (#2434)

parent 56fcd8e8
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
sgl-kernel/setup.py
View file @ 5de9a58e
...@@ -34,10 +34,12 @@ def rename_wheel(): ...@@ -34,10 +34,12 @@ def rename_wheel():
new_info = tmp_dir / f"sgl_kernel-{base_version}+cu{cuda_version}.dist-info" new_info = tmp_dir / f"sgl_kernel-{base_version}+cu{cuda_version}.dist-info"
old_info.rename(new_info) old_info.rename(new_info)
new_wheel = ( platform = "manylinux2014_x86_64"
wheel_dir new_wheel = wheel_dir / old_wheel.name.replace("linux_x86_64", platform)
/ f"sgl_kernel-{base_version}+cu{cuda_version}-{old_wheel.name.split('-', 2)[-1]}" 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: with zipfile.ZipFile(new_wheel, "w", zipfile.ZIP_DEFLATED) as new_zip:
for file_path in tmp_dir.rglob("*"): for file_path in tmp_dir.rglob("*"):
if file_path.is_file(): if file_path.is_file():
......
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