[CI] Add DeepGEMM warmup to stage-c-test-deepep-4-gpu (#19806)

Co-authored-by: Alison Shao <alisonshao@Mac.attlocal.net>
This commit is contained in:
Alison Shao
2026-03-03 16:18:12 -08:00
committed by GitHub
parent 441045a7bf
commit c18cff4f90

View File

@@ -1482,6 +1482,18 @@ jobs:
run: |
CUSTOM_BUILD_SGL_KERNEL=${{needs.check-changes.outputs.sgl_kernel}} bash scripts/ci/cuda/ci_install_deepep.sh
- name: Warmup DeepGEMM JIT Compilation
timeout-minutes: 25
run: |
python3 scripts/ci/cuda/warmup_deep_gemm.py \
lmsys/sglang-ci-dsv3-test:4
- name: Warmup Server CUDA Graphs
timeout-minutes: 25
run: |
python3 scripts/ci/cuda/warmup_server.py \
lmsys/sglang-ci-dsv3-test:4
- name: Run test
timeout-minutes: 20
run: |