ci: fix cu129 wheel tagging + pipefail-abort in install script (follow-up to #23497) (#23587)

This commit is contained in:
Jia Guo
2026-04-23 14:52:58 -07:00
committed by GitHub
parent 17210350fd
commit 6428392b6f
2 changed files with 11 additions and 15 deletions
+3 -1
View File
@@ -1,5 +1,5 @@
#!/usr/bin/env bash
# Align CUDA wheel filenames (+cu124/+cu128/+cu130) with internal METADATA Version and
# Align CUDA wheel filenames (+cu124/+cu128/+cu129/+cu130) with internal METADATA Version and
# WHEEL tags after build (fixes pip "inconsistent version" when only the .whl name changed).
# Unpack → patch WHEEL/METADATA → wheel pack (RECORD regenerated; no hand-editing).
set -ex
@@ -11,6 +11,8 @@ detect_cuda_suffix() {
echo "+cu124"
elif ls /usr/local/ 2>/dev/null | grep -q "12.8"; then
echo "+cu128"
elif ls /usr/local/ 2>/dev/null | grep -q "12.9"; then
echo "+cu129"
elif ls /usr/local/ 2>/dev/null | grep -q "13.0"; then
echo "+cu130"
else