From 1698c2341b7e3b17fedef1ddb74a3ee43fb1a98a Mon Sep 17 00:00:00 2001 From: Alison Shao <54658187+alisonshao@users.noreply.github.com> Date: Wed, 10 Dec 2025 08:32:33 -0800 Subject: [PATCH] [CI] Reduce stage-b auto-partition from 4 to 2 (#14769) Co-authored-by: Liangsheng Yin --- .github/workflows/pr-test.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 5ad945d7c..cee3812a1 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -413,7 +413,7 @@ jobs: strategy: fail-fast: false matrix: - partition: [0, 1, 2, 3] + partition: [0, 1] steps: - name: Checkout code uses: actions/checkout@v4 @@ -434,7 +434,7 @@ jobs: timeout-minutes: 30 run: | cd test/ - python3 run_suite.py --hw cuda --suite stage-b-test-small-1-gpu --auto-partition-id ${{ matrix.partition }} --auto-partition-size 4 + python3 run_suite.py --hw cuda --suite stage-b-test-small-1-gpu --auto-partition-id ${{ matrix.partition }} --auto-partition-size 2 multimodal-gen-test-1-gpu: needs: [check-changes, call-gate, sgl-kernel-build-wheels]