[Fix] Fix cu130 sgl-kernel wheel renaming (#12173)
This commit is contained in:
@@ -20,6 +20,8 @@ for wheel in "${wheel_files[@]}"; do
|
|||||||
new_wheel="${intermediate_wheel/-cp${cp_version}/+cu124-cp${cp_version}}"
|
new_wheel="${intermediate_wheel/-cp${cp_version}/+cu124-cp${cp_version}}"
|
||||||
elif ls /usr/local/ | grep -q "12.8"; then
|
elif ls /usr/local/ | grep -q "12.8"; then
|
||||||
new_wheel="${intermediate_wheel/-cp${cp_version}/+cu128-cp${cp_version}}"
|
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
|
else
|
||||||
new_wheel="$intermediate_wheel"
|
new_wheel="$intermediate_wheel"
|
||||||
fi
|
fi
|
||||||
|
|||||||
Reference in New Issue
Block a user