[Fix] Fix cu130 sgl-kernel wheel renaming (#12173)

2025-10-26 22:44:05 -07:00
parent 09af0a7b5a
commit 75c09e1ffe
1 changed files with 2 additions and 0 deletions
--- a/sgl-kernel/rename_wheels.sh
+++ b/sgl-kernel/rename_wheels.sh
@@ -20,6 +20,8 @@ for wheel in "${wheel_files[@]}"; do
        new_wheel="${intermediate_wheel/-cp${cp_version}/+cu124-cp${cp_version}}"
    elif ls /usr/local/ | grep -q "12.8"; then
        new_wheel="${intermediate_wheel/-cp${cp_version}/+cu128-cp${cp_version}}"
+    elif ls /usr/local/ | grep -q "13.0"; then
+        new_wheel="${intermediate_wheel/-cp${cp_version}/+cu130-cp${cp_version}}"
    else
        new_wheel="$intermediate_wheel"
    fi