minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[AMD] Fix Timeout: stage-b-test-2-gpu-large-amd,stage-b-test-1-gpu-large-amd (#22228)

Browse Source 
Co-authored-by: HAI <hixiao@gmail.com>
This commit is contained in:
YC Yen-Ching Tseng
2026-04-10 22:55:44 -07:00
committed by GitHub
co-authored by HAI
parent cf1436d6ae
commit 3ce72252de
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test-amd.yml
+2 -2
View File
@@ -462,7 +462,7 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
runner: [linux-mi325-1gpu-sglang] runner: [linux-mi325-1gpu-sglang]
part: [0, 1] part: [0, 1, 2]
runs-on: ${{matrix.runner}} runs-on: ${{matrix.runner}}
steps: steps:
- name: Checkout code - name: Checkout code
@@ -484,7 +484,7 @@ jobs:
- name: Run test - name: Run test
timeout-minutes: 30 timeout-minutes: 30
run: | run: |
bash scripts/ci/amd/amd_ci_exec.sh -w "/sglang-checkout/test" python3 run_suite.py --hw amd --suite stage-b-test-1-gpu-large-amd --auto-partition-id ${{ matrix.part }} --auto-partition-size 2 --timeout-per-file 1800 ${{ inputs.continue_on_error && '--continue-on-error' || '' }} bash scripts/ci/amd/amd_ci_exec.sh -w "/sglang-checkout/test" python3 run_suite.py --hw amd --suite stage-b-test-1-gpu-large-amd --auto-partition-id ${{ matrix.part }} --auto-partition-size 3 --timeout-per-file 1800 ${{ inputs.continue_on_error && '--continue-on-error' || '' }}
stage-b-test-2-gpu-large-amd: stage-b-test-2-gpu-large-amd:
needs: [check-changes, stage-a-test-1-gpu-small-amd] needs: [check-changes, stage-a-test-1-gpu-small-amd]
test/registered/perf/test_bench_serving_2gpu.py
+1 -1
View File
@@ -15,7 +15,7 @@ from sglang.test.test_utils import (
) )
register_cuda_ci(est_time=660, suite="stage-b-test-2-gpu-large") register_cuda_ci(est_time=660, suite="stage-b-test-2-gpu-large")
register_amd_ci(est_time=1100, suite="stage-b-test-2-gpu-large-amd") register_amd_ci(est_time=1450, suite="stage-b-test-2-gpu-large-amd")
class TestBenchServing2GPU(CustomTestCase): class TestBenchServing2GPU(CustomTestCase):
test/registered/rl/test_update_weights_from_distributed.py
+1 -1
View File
@@ -44,7 +44,7 @@ from sglang.test.test_utils import (
from sglang.utils import terminate_process from sglang.utils import terminate_process
register_cuda_ci(est_time=91, suite="stage-b-test-2-gpu-large") register_cuda_ci(est_time=91, suite="stage-b-test-2-gpu-large")
register_amd_ci(est_time=103, suite="stage-b-test-2-gpu-large-amd") register_amd_ci(est_time=400, suite="stage-b-test-2-gpu-large-amd")
mp.set_start_method("spawn", force=True) mp.set_start_method("spawn", force=True)