From 5de9a58eca5fe9ce6ed4ad6f09efe5f2cecbab6e Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Wed, 11 Dec 2024 06:17:41 +0800 Subject: [PATCH] fix: use manylinux2014_x86_64 tag (#2434) --- sgl-kernel/setup.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py index aaaae62be..7672eb094 100644 --- a/sgl-kernel/setup.py +++ b/sgl-kernel/setup.py @@ -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():