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

Add a special label for b200 CI runner that can run kernel tests (#15033)

Browse Source
This commit is contained in:
Kangyan-Zhou
2025-12-12 19:41:36 -08:00
committed by GitHub
parent 665cb02003
commit 291396544e
1 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test.yml
+16 -4
View File
@@ -37,6 +37,7 @@ jobs:
sgl_kernel: ${{ steps.filter.outputs.sgl_kernel || steps.scheduled.outputs.sgl_kernel }}
multimodal_gen: ${{ steps.filter.outputs.multimodal_gen || steps.scheduled.outputs.multimodal_gen }}
max_parallel: ${{ steps.set-parallel.outputs.max_parallel }}
b200_runner: ${{ steps.set-runner.outputs.b200_runner }}
steps:
- name: Checkout code
uses: actions/checkout@v4
@@ -80,6 +81,16 @@ jobs:
echo "Using default max_parallel of 8"
fi
- name: Set B200 runner tag
id: set-runner
run: |
sgl_kernel="${{ steps.filter.outputs.sgl_kernel || steps.scheduled.outputs.sgl_kernel }}"
if [[ "$sgl_kernel" == "true" ]]; then
echo "b200_runner=4-gpu-b200-kernel" >> $GITHUB_OUTPUT
else
echo "b200_runner=4-gpu-b200" >> $GITHUB_OUTPUT
fi
- name: Show filter results in summary (table)
run: |
{
@@ -91,6 +102,7 @@ jobs:
echo "| sgl_kernel | ${{ steps.filter.outputs.sgl_kernel || steps.scheduled.outputs.sgl_kernel }} |"
echo "| multimodal_gen | ${{ steps.filter.outputs.multimodal_gen || steps.scheduled.outputs.multimodal_gen }} |"
echo "| max_parallel | ${{ steps.set-parallel.outputs.max_parallel }} |"
echo "| b200_runner | ${{ steps.set-runner.outputs.b200_runner }} |"
} >> $GITHUB_STEP_SUMMARY
# =============================================== PR Gate ====================================================
@@ -295,9 +307,9 @@ jobs:
sgl-kernel-b200-test:
needs: [check-changes, sgl-kernel-build-wheels]
if: needs.check-changes.outputs.sgl_kernel == 'true'
runs-on: 4-gpu-b200
runs-on: ${{ needs.check-changes.outputs.b200_runner }}
env:
RUNNER_LABELS: 4-gpu-b200
RUNNER_LABELS: ${{ needs.check-changes.outputs.b200_runner }}
steps:
- uses: actions/checkout@v4
@@ -1186,9 +1198,9 @@ jobs:
((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true'))
)
)
runs-on: 4-gpu-b200
runs-on: ${{ needs.check-changes.outputs.b200_runner }}
env:
RUNNER_LABELS: 4-gpu-b200
RUNNER_LABELS: ${{ needs.check-changes.outputs.b200_runner }}
strategy:
fail-fast: false
matrix: