Add Mistral Large 3 basic test to PR CI (#14460)

This commit is contained in:
Alison Shao
2025-12-05 10:38:52 -08:00
committed by GitHub
parent cf9a774cf8
commit 16e8463a90
3 changed files with 117 additions and 1 deletions

View File

@@ -1203,6 +1203,32 @@ jobs:
cd test/srt
python3 run_suite.py --suite per-commit-4-gpu-gb200 --auto-partition-id 0 --auto-partition-size 1 --timeout-per-file 3600
unit-test-backend-8-gpu-b200:
needs: [check-changes, call-gate, unit-test-backend-2-gpu]
if: |
(inputs.target_stage == 'unit-test-backend-8-gpu-b200') ||
(
always() &&
(github.event_name == 'schedule' || (!failure() && !cancelled())) &&
((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true'))
)
runs-on: 8-gpu-b200
env:
RUNNER_LABELS: 8-gpu-b200
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Install dependencies
run: |
IS_BLACKWELL=1 bash scripts/ci/ci_install_dependency.sh
- name: Run test
timeout-minutes: 45
run: |
cd test/srt
python3 run_suite.py --suite per-commit-8-gpu-b200 --timeout-per-file 1800
pr-test-finish:
needs:
[
@@ -1235,6 +1261,7 @@ jobs:
unit-test-deepep-8-gpu,
unit-test-backend-4-gpu-b200,
unit-test-backend-4-gpu-gb200,
unit-test-backend-8-gpu-b200,
]
if: always()
runs-on: ubuntu-latest