EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

fix setup for sgl kernel (#2917)

Browse Source
This commit is contained in:
Yineng Zhang
2025-01-16 18:17:34 +08:00
committed by GitHub
parent bf8d07a6f9
commit 2dc957d421
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
sgl-kernel/setup.py
View File

@@ -15,11 +15,12 @@ def get_version():
def update_wheel_platform_tag(): def update_wheel_platform_tag():
wheel_dir = Path("dist") wheel_dir = Path("dist")
old_wheel = next(wheel_dir.glob("*.whl")) if wheel_dir.exists() and wheel_dir.is_dir():
new_wheel = wheel_dir / old_wheel.name.replace( old_wheel = next(wheel_dir.glob("*.whl"))
"linux_x86_64", "manylinux2014_x86_64" new_wheel = wheel_dir / old_wheel.name.replace(
) "linux_x86_64", "manylinux2014_x86_64"
old_wheel.rename(new_wheel) )
old_wheel.rename(new_wheel)
cutlass = root / "3rdparty" / "cutlass" cutlass = root / "3rdparty" / "cutlass"