diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py index ffdf5539b..cdf8bb2a0 100644 --- a/sgl-kernel/setup.py +++ b/sgl-kernel/setup.py @@ -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()