[AMD][CI] Publish ROCm 10 release images and kernel wheel (#38763)
This commit is contained in:
+9
-1
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user