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

CI: split JIT kernel unit tests into two partitions (#36775)

Browse Source
This commit is contained in:
Po-Han Huang (NVIDIA)
2026-08-28 09:32:58 -07:00
committed by GitHub
parent 9579bff860
commit 6ff2fe6a6a
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test-jit-kernel.yml
+8 -1
View File
@@ -47,6 +47,10 @@ jobs:
# Runs whenever call-jit-kernel-tests dispatches this workflow. That caller is the # Runs whenever call-jit-kernel-tests dispatches this workflow. That caller is the
# single gate (PR jit_kernel changes, or scheduled/parallel-dispatch full runs), so # single gate (PR jit_kernel changes, or scheduled/parallel-dispatch full runs), so
# the sub-jobs no longer re-exclude schedule/parallel-dispatch here. # the sub-jobs no longer re-exclude schedule/parallel-dispatch here.
strategy:
fail-fast: false
matrix:
partition: [0, 1]
runs-on: 1-gpu-h100 runs-on: 1-gpu-h100
timeout-minutes: 240 timeout-minutes: 240
steps: steps:
@@ -85,7 +89,10 @@ jobs:
timeout-minutes: 30 timeout-minutes: 30
run: | run: |
cd test/ cd test/
python3 run_suite.py --hw cuda --suite base-b-kernel-unit-test-1-gpu-large python3 run_suite.py --hw cuda \
--suite base-b-kernel-unit-test-1-gpu-large \
--auto-partition-id ${{ matrix.partition }} \
--auto-partition-size 2
jit-kernel-multigpu-unit-test: jit-kernel-multigpu-unit-test:
# Runs whenever call-jit-kernel-tests dispatches this workflow. That caller is the # Runs whenever call-jit-kernel-tests dispatches this workflow. That caller is the