feat: add cu128 identifier for sgl-kernel (#5287)

This commit is contained in:
Yineng Zhang
2025-04-11 01:58:46 -07:00
committed by GitHub
parent 7074e9ca20
commit b75275b6f2
3 changed files with 8 additions and 5 deletions

View File

@@ -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."