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

ci: adjust partition counts for stage-b and unit tests (#16617)

Browse Source
This commit is contained in:
Alison Shao
2026-01-06 20:18:10 -08:00
committed by GitHub
parent d8b8198192
commit 38895a0064
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test.yml
+4 -4
View File
@@ -587,7 +587,7 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
partition: [0, 1, 2, 3, 4, 5, 6, 7] partition: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
steps: steps:
- name: Checkout code - name: Checkout code
uses: actions/checkout@v4 uses: actions/checkout@v4
@@ -614,7 +614,7 @@ jobs:
if [[ "${{ needs.check-changes.outputs.continue_on_error }}" == "true" ]]; then if [[ "${{ needs.check-changes.outputs.continue_on_error }}" == "true" ]]; then
CONTINUE_ON_ERROR_FLAG="--continue-on-error" CONTINUE_ON_ERROR_FLAG="--continue-on-error"
fi fi
python3 run_suite.py --hw cuda --suite stage-b-test-small-1-gpu --auto-partition-id ${{ matrix.partition }} --auto-partition-size 8 $CONTINUE_ON_ERROR_FLAG python3 run_suite.py --hw cuda --suite stage-b-test-small-1-gpu --auto-partition-id ${{ matrix.partition }} --auto-partition-size 11 $CONTINUE_ON_ERROR_FLAG
stage-b-test-large-1-gpu: stage-b-test-large-1-gpu:
needs: [check-changes, call-gate, stage-a-test-1, sgl-kernel-build-wheels] needs: [check-changes, call-gate, stage-a-test-1, sgl-kernel-build-wheels]
@@ -946,7 +946,7 @@ jobs:
fail-fast: false fail-fast: false
max-parallel: ${{ fromJson(needs.check-changes.outputs.max_parallel) }} max-parallel: ${{ fromJson(needs.check-changes.outputs.max_parallel) }}
matrix: matrix:
part: [0, 1, 2, 3, 4, 5] part: [0, 1]
steps: steps:
- name: Checkout code - name: Checkout code
uses: actions/checkout@v4 uses: actions/checkout@v4
@@ -977,7 +977,7 @@ jobs:
if [[ "${{ needs.check-changes.outputs.continue_on_error }}" == "true" ]]; then if [[ "${{ needs.check-changes.outputs.continue_on_error }}" == "true" ]]; then
CONTINUE_ON_ERROR_FLAG="--continue-on-error" CONTINUE_ON_ERROR_FLAG="--continue-on-error"
fi fi
python3 run_suite.py --suite per-commit-1-gpu --auto-partition-id ${{ matrix.part }} --auto-partition-size 6 $RETRY_FLAG $CONTINUE_ON_ERROR_FLAG python3 run_suite.py --suite per-commit-1-gpu --auto-partition-id ${{ matrix.part }} --auto-partition-size 2 $RETRY_FLAG $CONTINUE_ON_ERROR_FLAG
stage-b-test-4-gpu-b200: stage-b-test-4-gpu-b200: