feat: add cu128 identifier for sgl-kernel (#5287)
This commit is contained in:
@@ -5,13 +5,17 @@ WHEEL_DIR="dist"
|
||||
|
||||
wheel_files=($WHEEL_DIR/*.whl)
|
||||
for wheel in "${wheel_files[@]}"; do
|
||||
new_wheel="${wheel/linux/manylinux2014}"
|
||||
intermediate_wheel="${wheel/linux/manylinux2014}"
|
||||
|
||||
if ls /usr/local/ | grep -q "12.8"; then
|
||||
new_wheel="${intermediate_wheel/-cp39/+cu128-cp39}"
|
||||
else
|
||||
new_wheel="$intermediate_wheel"
|
||||
fi
|
||||
|
||||
if [[ "$wheel" != "$new_wheel" ]]; then
|
||||
echo "Renaming $wheel to $new_wheel"
|
||||
mv -- "$wheel" "$new_wheel"
|
||||
|
||||
fi
|
||||
done
|
||||
|
||||
echo "Wheel renaming completed."
|
||||
|
||||
Reference in New Issue
Block a user