[CI] Fix rerun-ut workflow: add DeepEP install, RDMA env, Blackwell detection (#19803)

This commit is contained in:
Liangsheng Yin
2026-03-03 15:17:16 -08:00
committed by GitHub
parent 753da27535
commit c7ffbf25e9
2 changed files with 28 additions and 6 deletions

View File

@@ -17,10 +17,16 @@ on:
required: false
type: string
default: ""
use_deepep:
description: "Use ci_install_deepep.sh instead of ci_install_dependency.sh"
required: false
type: string
default: "false"
env:
SGLANG_IS_IN_CI: true
SGLANG_CUDA_COREDUMP: "1"
SGLANG_JIT_DEEPGEMM_FAST_WARMUP: true
permissions:
actions: write
@@ -32,7 +38,8 @@ jobs:
timeout-minutes: 120
env:
RUNNER_LABELS: ${{ inputs.runner_label }}
IS_BLACKWELL: ${{ inputs.runner_label == '1-gpu-5090' && '1' || '' }}
IS_BLACKWELL: ${{ (inputs.runner_label == '1-gpu-5090' || contains(inputs.runner_label, 'b200')) && '1' || '' }}
SGLANG_CI_RDMA_ALL_DEVICES: ${{ inputs.runner_label == '8-gpu-h20' && 'mlx5_1,mlx5_2,mlx5_3,mlx5_4' || '' }}
steps:
- name: Checkout code
uses: actions/checkout@v4
@@ -45,7 +52,11 @@ jobs:
if [[ "${{ inputs.runner_label }}" == "1-gpu-5090" ]]; then
source /etc/profile.d/sglang-ci.sh
fi
bash scripts/ci/cuda/ci_install_dependency.sh
if [[ "${{ inputs.use_deepep }}" == "true" ]]; then
bash scripts/ci/cuda/ci_install_deepep.sh
else
bash scripts/ci/cuda/ci_install_dependency.sh
fi
- name: Run test
timeout-minutes: 60