[AMD] Build Miles nightly ROCm images with docker/build.py and test ROCm 7.2 (#31757)

This commit is contained in:
Xinyu Jiang
2026-07-23 08:30:09 +08:00
committed by GitHub
parent 24da0e51b6
commit 8ce68370b5
3 changed files with 39 additions and 51 deletions
@@ -14,13 +14,13 @@ jobs:
if: github.repository == 'sgl-project/sglang'
runs-on: amd-docker-scale
environment: "prod"
permissions:
contents: read
strategy:
fail-fast: false
matrix:
include:
- gpu_arch: gfx950
rocm_tag: rocm720-mi35x
sglang_image_tag: v0.5.10-rocm720-mi35x
variant:
- rocm720-mi35x
steps:
- name: Checkout miles repository
uses: actions/checkout@v4
@@ -28,13 +28,21 @@ jobs:
repository: radixark/miles
ref: main
fetch-depth: 0
persist-credentials: false
- name: Record miles commit
run: echo "Building from miles main @ $(git rev-parse HEAD)"
- name: Set date
run: echo "DATE=$(date +%Y%m%d)" >> $GITHUB_ENV
- name: Record miles commit
id: miles
run: echo "sha=$(git rev-parse HEAD)" >> $GITHUB_OUTPUT
- name: Install docker/build.py dependencies
run: python3 -m pip install --quiet typer
- name: Build image via miles docker/build.py and tag with date
run: |
python3 docker/build.py --variant ${{ matrix.variant }} --image-tag custom --custom-tag miles
docker tag rocm/sgl-dev:miles-${{ matrix.variant }} rocm/sgl-dev:miles-${{ matrix.variant }}-${DATE}
- name: Login to Docker Hub (AMD)
uses: docker/login-action@v2
@@ -42,17 +50,5 @@ jobs:
username: ${{ secrets.DOCKERHUB_AMD_USERNAME }}
password: ${{ secrets.DOCKERHUB_AMD_TOKEN }}
- name: Build and push miles image to rocm/sgl-dev
run: |
base="miles-${{ matrix.rocm_tag }}"
dated="${base}-${{ env.DATE }}"
echo "IMAGE_TAG=${dated}" >> $GITHUB_ENV
echo "Building rocm/sgl-dev:${dated} from miles main @ ${{ steps.miles.outputs.sha }}"
docker build . \
-f docker/Dockerfile.rocm \
--build-arg GPU_ARCH=${{ matrix.gpu_arch }} \
--build-arg SGLANG_IMAGE_TAG=${{ matrix.sglang_image_tag }} \
--build-arg MILES_COMMIT=${{ steps.miles.outputs.sha }} \
-t rocm/sgl-dev:${dated} \
--no-cache
docker push rocm/sgl-dev:${dated}
- name: Push dated tag
run: docker push rocm/sgl-dev:miles-${{ matrix.variant }}-${DATE}