[AMD][CI] Publish ROCm 10 release images and kernel wheel (#38763)

This commit is contained in:
Michael
2026-09-09 23:57:00 -07:00
committed by GitHub
parent a1ec35a330
commit f3152439d0
4 changed files with 138 additions and 8 deletions
+9 -1
View File
@@ -3,6 +3,11 @@ set -ex
WHEEL_DIR="dist"
# ROCm version for the wheel's local version tag (e.g. 720 -> +rocm720). It is
# read from the /opt/rocm-<version> tree when the caller does not name one; the
# pip-installed ROCm 10 SDK has no such tree, so that flavor passes it in.
ROCM_WHEEL_VERSION="${1:-}"
wheel_files=($WHEEL_DIR/*.whl)
for wheel in "${wheel_files[@]}"; do
intermediate_wheel="${wheel/linux/manylinux2014}"
@@ -17,7 +22,10 @@ for wheel in "${wheel_files[@]}"; do
fi
# Detect ROCm version and add appropriate suffix
ver_abrv=$(realpath /opt/rocm-* | sed -e 's/.*-//' -e 's/\.//g')
ver_abrv="${ROCM_WHEEL_VERSION}"
if [[ -z "$ver_abrv" ]]; then
ver_abrv=$(realpath /opt/rocm-* | sed -e 's/.*-//' -e 's/\.//g')
fi
new_wheel=${intermediate_wheel/-cp${cp_version}/+rocm${ver_abrv}-cp${cp_version}}
if [[ "$wheel" != "$new_wheel" ]]; then