minor: update pypi tag (#2463)
This commit is contained in:
@@ -49,6 +49,15 @@ def rename_wheel():
|
|||||||
shutil.rmtree(tmp_dir)
|
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(
|
setup(
|
||||||
name="sgl-kernel",
|
name="sgl-kernel",
|
||||||
version=get_version(),
|
version=get_version(),
|
||||||
@@ -78,3 +87,5 @@ setup(
|
|||||||
cmdclass={"build_ext": BuildExtension},
|
cmdclass={"build_ext": BuildExtension},
|
||||||
install_requires=["torch"],
|
install_requires=["torch"],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
update_wheel_platform_tag()
|
||||||
|
|||||||
Reference in New Issue
Block a user