Skip to content

GitLab

Unverified Commit 2dc957d4 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix setup for sgl kernel (#2917)

parent bf8d07a6
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 5 deletions
sgl-kernel/setup.py
View file @ 2dc957d4
...@@ -15,11 +15,12 @@ def get_version(): ...@@ -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"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment