diff --git a/.github/workflows/pr-test-amd.yml b/.github/workflows/pr-test-amd.yml index 67ab10853..6f0de0439 100644 --- a/.github/workflows/pr-test-amd.yml +++ b/.github/workflows/pr-test-amd.yml @@ -462,7 +462,7 @@ jobs: fail-fast: false matrix: runner: [linux-mi325-1gpu-sglang] - part: [0, 1] + part: [0, 1, 2] runs-on: ${{matrix.runner}} steps: - name: Checkout code @@ -484,7 +484,7 @@ jobs: - name: Run test timeout-minutes: 30 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: needs: [check-changes, stage-a-test-1-gpu-small-amd] diff --git a/test/registered/perf/test_bench_serving_2gpu.py b/test/registered/perf/test_bench_serving_2gpu.py index 7da00c13f..d03cf2499 100644 --- a/test/registered/perf/test_bench_serving_2gpu.py +++ b/test/registered/perf/test_bench_serving_2gpu.py @@ -15,7 +15,7 @@ from sglang.test.test_utils import ( ) 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): diff --git a/test/registered/rl/test_update_weights_from_distributed.py b/test/registered/rl/test_update_weights_from_distributed.py index 3d464db3f..d728fbc53 100644 --- a/test/registered/rl/test_update_weights_from_distributed.py +++ b/test/registered/rl/test_update_weights_from_distributed.py @@ -44,7 +44,7 @@ from sglang.test.test_utils import ( from sglang.utils import terminate_process 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)