[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
+17 -3
View File
@@ -15,6 +15,7 @@ on:
default: 'all'
options:
- 'all'
- rocm10
- rocm724
- rocm720
- rocm700
@@ -25,11 +26,11 @@ jobs:
runs-on: amd-docker-scale
environment: 'prod'
strategy:
# Six flavors publish independently: a failure in one must not cancel the
# others mid-push and leave the release tag with a partial image set.
# Eight flavors publish independently: a failure in one must not cancel
# the others mid-push and leave the release tag with a partial image set.
fail-fast: false
matrix:
rocm_version: ${{ fromJson((github.event_name == 'workflow_dispatch' && inputs.rocm_version != 'all' && inputs.rocm_version != '') && format('["{0}"]', inputs.rocm_version) || '["rocm700", "rocm720", "rocm724"]') }}
rocm_version: ${{ fromJson((github.event_name == 'workflow_dispatch' && inputs.rocm_version != 'all' && inputs.rocm_version != '') && format('["{0}"]', inputs.rocm_version) || '["rocm700", "rocm720", "rocm724", "rocm10"]') }}
gpu_arch: ['gfx942', 'gfx950']
build_type: ['all']
steps:
@@ -89,6 +90,19 @@ jobs:
echo "Unsupported gfx arch"
exit 1
fi
elif [ "${{ matrix.rocm_version }}" = "rocm10" ]; then
# The Dockerfile stages carry the full SDK version (gfx942-rocm1000)
# while the published tag keeps the rocm10 flavor name that CI and
# the nightly images already use.
gpu_arch_suffix="-rocm1000"
if [ "${{ matrix.gpu_arch }}" = "gfx942" ]; then
rocm_tag="rocm10-mi30x"
elif [ "${{ matrix.gpu_arch }}" = "gfx950" ]; then
rocm_tag="rocm10-mi35x"
else
echo "Unsupported gfx arch"
exit 1
fi
else
echo "Unsupported rocm version"
exit 1