Skip to content

Commit bbb8d94

Browse files
alisonshaoAlison Shao
authored andcommitted
[CI] Add DeepGEMM warmup to stage-c-test-deepep-4-gpu (sgl-project#19806)
Co-authored-by: Alison Shao <alisonshao@Mac.attlocal.net>
1 parent 205fc7c commit bbb8d94

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

.github/workflows/pr-test.yml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1482,6 +1482,18 @@ jobs:
14821482
run: |
14831483
CUSTOM_BUILD_SGL_KERNEL=${{needs.check-changes.outputs.sgl_kernel}} bash scripts/ci/cuda/ci_install_deepep.sh
14841484
1485+
- name: Warmup DeepGEMM JIT Compilation
1486+
timeout-minutes: 25
1487+
run: |
1488+
python3 scripts/ci/cuda/warmup_deep_gemm.py \
1489+
lmsys/sglang-ci-dsv3-test:4
1490+
1491+
- name: Warmup Server CUDA Graphs
1492+
timeout-minutes: 25
1493+
run: |
1494+
python3 scripts/ci/cuda/warmup_server.py \
1495+
lmsys/sglang-ci-dsv3-test:4
1496+
14851497
- name: Run test
14861498
timeout-minutes: 20
14871499
run: |

0 commit comments

Comments
 (0)