From 2dc957d421fdd6a92b04ba52e9e0c745f1e370b1 Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Thu, 16 Jan 2025 18:17:34 +0800 Subject: [PATCH] fix setup for sgl kernel (#2917) --- sgl-kernel/setup.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py index da6b465d8..36596a1b0 100644 --- a/sgl-kernel/setup.py +++ b/sgl-kernel/setup.py @@ -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"