minor: update pypi tag (#2463)

This commit is contained in:
Yineng Zhang
2024-12-12 15:21:45 +08:00
committed by GitHub
parent 32ed016041
commit dedaf8cd48

View File

@@ -49,6 +49,15 @@ def rename_wheel():
shutil.rmtree(tmp_dir)
def update_wheel_platform_tag():
wheel_dir = Path("dist")
old_wheel = next(wheel_dir.glob("*.whl"))
new_wheel = wheel_dir / old_wheel.name.replace(
"linux_x86_64", "manylinux2014_x86_64"
)
old_wheel.rename(new_wheel)
setup(
name="sgl-kernel",
version=get_version(),
@@ -78,3 +87,5 @@ setup(
cmdclass={"build_ext": BuildExtension},
install_requires=["torch"],
)
update_wheel_platform_tag()