EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Update amd docker image. (#8045)

Browse Source 
Co-authored-by: Hubert Lu <55214931+hubertlu-tw@users.noreply.github.com>
This commit is contained in:
Sai Enduri
2025-07-15 15:09:56 -07:00
committed by GitHub
parent 14f1f1514b
commit f06bd210c0
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
.github/workflows/pr-test-amd.yml vendored
View File

@@ -223,7 +223,7 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
runner: [linux-mi300-gpu-1, linux-mi325-gpu-1] runner: [linux-mi300-gpu-1, linux-mi325-gpu-1]
part: [0, 1, 2, 3, 4, 5] part: [0, 1, 2, 3, 4, 5, 6]
runs-on: ${{matrix.runner}} runs-on: ${{matrix.runner}}
steps: steps:
- name: Checkout code - name: Checkout code
@@ -238,9 +238,9 @@ jobs:
run: bash scripts/amd_ci_install_dependency.sh run: bash scripts/amd_ci_install_dependency.sh
- name: Run test - name: Run test
timeout-minutes: 40 timeout-minutes: 50
run: | run: |
bash scripts/amd_ci_exec.sh python3 run_suite.py --suite per-commit-amd --auto-partition-id ${{ matrix.part }} --auto-partition-size 6 bash scripts/amd_ci_exec.sh python3 run_suite.py --suite per-commit-amd --auto-partition-id ${{ matrix.part }} --auto-partition-size 7
unit-test-backend-2-gpu-amd: unit-test-backend-2-gpu-amd:
if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') && if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&
@@ -293,7 +293,7 @@ jobs:
- name: Run CustomAllReduce test - name: Run CustomAllReduce test
timeout-minutes: 10 timeout-minutes: 10
run: | run: |
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 bash scripts/amd_ci_exec.sh python3 -m unittest test_custom_allreduce.TestCustomAllReduce bash scripts/amd_ci_exec.sh -e CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python3 -m unittest test_custom_allreduce.TestCustomAllReduce
unit-test-sgl-kernel-amd: unit-test-sgl-kernel-amd:
if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') && if: (github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request') &&

2
scripts/amd_ci_start_container.sh
View File

@@ -9,7 +9,7 @@ else
fi fi
# Pull the image # Pull the image
IMAGE="ghcr.io/saienduri/sglang:aiter-1.3" IMAGE="rocm/sgl-dev:v0.4.9.post2-rocm630-mi30x-20250715"
echo "Pulling Docker image: $IMAGE" echo "Pulling Docker image: $IMAGE"
docker pull "$IMAGE" docker pull "$IMAGE"