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

Co-authored-by: HAI <hixiao@gmail.com>

.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]

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):

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)