fix setup for sgl kernel (#2917)

This commit is contained in:
Yineng Zhang
2025-01-16 18:17:34 +08:00
committed by GitHub
parent bf8d07a6f9
commit 2dc957d421

View File

@@ -15,11 +15,12 @@ def get_version():
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)
if wheel_dir.exists() and wheel_dir.is_dir():
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)
cutlass = root / "3rdparty" / "cutlass"