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

ci: run Stage A CUDA tests as stage-a-test-small-1-gpu on 5090 (#20988)

Browse Source
This commit is contained in:
Lianmin Zheng
2026-03-20 02:55:16 -07:00
committed by GitHub
parent fa89d152c0
commit c1da420799
12 changed files with 78 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/ci/utils/slash_command_handler.py
View File

@@ -253,7 +253,7 @@ def handle_rerun_stage(
# Valid NVIDIA stage names that support target_stage
nvidia_stages = [
"stage-a-test-1",
"stage-a-test-small-1-gpu",
"stage-a-cpu-only",
"stage-b-test-small-1-gpu",
"stage-b-test-large-1-gpu",
@@ -274,7 +274,7 @@ def handle_rerun_stage(
amd_stages = [
"sgl-kernel-unit-test-amd",
"sgl-kernel-unit-test-2-gpu-amd",
"stage-a-test-1-amd",
"stage-a-test-small-1-gpu-amd",
"stage-b-test-small-1-gpu-amd",
"stage-b-test-small-1-gpu-amd-nondeterministic",
"stage-b-test-small-1-gpu-amd-mi35x",
@@ -334,7 +334,10 @@ def handle_rerun_stage(
f"Triggering {workflow_name} workflow on ref: {ref}, PR head SHA: {pr_head_sha}"
)
if is_amd_stage:
inputs = {"target_stage": stage_name, "pr_head_sha": pr_head_sha}
inputs = {
"target_stage": stage_name,
"pr_head_sha": pr_head_sha,
}
else:
inputs = {
"version": "release",
@@ -410,7 +413,7 @@ def handle_rerun_stage(
CUDA_SUITE_TO_RUNNER = {
"stage-a-test-1": "1-gpu-runner",
"stage-a-test-small-1-gpu": "1-gpu-5090",
"stage-a-cpu-only": "ubuntu-latest",
"stage-b-test-small-1-gpu": "1-gpu-5090",
"stage-b-test-large-1-gpu": "1-gpu-runner",