diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index 1d71b283e..40c2a353e 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -656,7 +656,7 @@ jobs: fail-fast: false max-parallel: ${{ fromJson(needs.check-changes.outputs.max_parallel) }} matrix: - partition: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] + partition: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11] steps: - name: Checkout code uses: actions/checkout@v4 @@ -684,7 +684,7 @@ jobs: if [[ "${{ needs.check-changes.outputs.continue_on_error }}" == "true" ]]; then CONTINUE_ON_ERROR_FLAG="--continue-on-error" fi - python3 run_suite.py --hw cuda --suite stage-b-test-large-1-gpu --auto-partition-id ${{ matrix.partition }} --auto-partition-size 10 $CONTINUE_ON_ERROR_FLAG + python3 run_suite.py --hw cuda --suite stage-b-test-large-1-gpu --auto-partition-id ${{ matrix.partition }} --auto-partition-size 12 $CONTINUE_ON_ERROR_FLAG stage-b-test-large-2-gpu: needs: [check-changes, call-gate, stage-a-test-1, sgl-kernel-build-wheels]