[CI] Add 5090 GPU test workflow with compatible tests only #53
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: 5090 GPU Test | |
| on: | |
| pull_request: | |
| branches: [main] | |
| paths: | |
| - '.github/workflows/test-5090.yml' | |
| workflow_dispatch: | |
| concurrency: | |
| group: 5090-test-${{ github.ref }} | |
| cancel-in-progress: true | |
| env: | |
| RUNNER_LABELS: 1-gpu-5090 | |
| SGLANG_IS_IN_CI: "true" | |
| jobs: | |
| # Run legacy stage-b-test-small-1-gpu-5090 suite (13 tests) | |
| # This tests the 13 tests that have explicit 5090 registration | |
| stage-b-test-small-1-gpu-5090: | |
| if: github.repository == 'sgl-project/sglang' | |
| runs-on: 1-gpu-5090 | |
| continue-on-error: true | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| partition: [0, 1, 2, 3] | |
| steps: | |
| - name: Checkout code | |
| uses: actions/checkout@v4 | |
| - name: Install dependencies | |
| timeout-minutes: 15 | |
| env: | |
| IS_BLACKWELL: "1" | |
| run: | | |
| source /etc/profile.d/sglang-ci.sh | |
| bash scripts/ci/ci_install_dependency.sh | |
| - name: Run stage-b-test-small-1-gpu-5090 | |
| timeout-minutes: 60 | |
| run: | | |
| source /etc/profile.d/sglang-ci.sh | |
| cd test/ | |
| python3 run_suite.py --hw cuda --suite stage-b-test-small-1-gpu-5090 \ | |
| --auto-partition-id ${{ matrix.partition }} \ | |
| --auto-partition-size 4 \ | |
| --continue-on-error | |
| summary: | |
| needs: [stage-b-test-small-1-gpu-5090] | |
| if: always() | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Summary | |
| run: | | |
| echo "## 5090 GPU Test Results (13 Legacy Tests)" >> $GITHUB_STEP_SUMMARY | |
| echo "" >> $GITHUB_STEP_SUMMARY | |
| echo "Testing 13 tests with explicit stage-b-test-small-1-gpu-5090 registration:" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_torchao.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_mamba_unittest.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_radix_cache_unit.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_hidden_states.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_mamba_ssm.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_mamba_ssm_ssd.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_causal_conv1d.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_external_models.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_build_eagle_tree.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_hicache_storage.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_evs.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_lora_hf_sgl_logprob_diff.py" >> $GITHUB_STEP_SUMMARY | |
| echo "- test_update_weights_from_tensor.py" >> $GITHUB_STEP_SUMMARY | |
| echo "" >> $GITHUB_STEP_SUMMARY | |
| echo "| Test Suite | Status |" >> $GITHUB_STEP_SUMMARY | |
| echo "|------------|--------|" >> $GITHUB_STEP_SUMMARY | |
| echo "| stage-b-test-small-1-gpu-5090 | ${{ needs.stage-b-test-small-1-gpu-5090.result }} |" >> $GITHUB_STEP_SUMMARY |