diff --git a/sgl-kernel/rename_wheels.sh b/sgl-kernel/rename_wheels.sh index 018eeb27b..915f069e6 100755 --- 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