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

Add GB300 base C CI suite (#27427)

Browse Source
This commit is contained in:
Baizhou Zhang
2026-06-06 02:27:30 -07:00
committed by GitHub
parent 42fe025280
commit 9a48bf75f5
9 changed files with 89 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/ci/utils/slash_command_handler.py
+19 -9
View File
@@ -709,8 +709,9 @@ def _extract_legacy_suites(content):
# matches the runner the nightly/weekly pipeline actually uses (see
# .github/workflows/{nightly,weekly}-test-nvidia.yml), so /rerun-test can still
# dispatch a single nightly/weekly test. The runner label, install script,
# timeout and rdma_devices are then resolved from runner_configs.yml as usual,
# keeping that file the single source of truth for runner details.
# timeout, grace_blackwell, and rdma_devices are then resolved from
# runner_configs.yml as usual, keeping that file the single source of truth for
# runner details.
#
# Suites on hardware with no matching runner_config (e.g. nightly-4-gpu-gb300)
# and non-CUDA suites (npu/amd) are intentionally absent and stay
@@ -740,6 +741,7 @@ def _dispatch_err(suite, msg):
"runner_label": None,
"install_script": "",
"install_timeout": "",
"grace_blackwell": "0",
"rdma_devices": "",
"is_cpu": False,
"error": msg,
@@ -780,6 +782,7 @@ def _resolve_runner_config(rc, full_path, suite):
"runner_label": runs_on,
"install_script": install_script,
"install_timeout": str(cfg["install_timeout"]),
"grace_blackwell": str(cfg.get("grace_blackwell", "0")),
"rdma_devices": cfg.get("rdma_devices", ""),
"is_cpu": False,
"error": None,
@@ -793,9 +796,9 @@ def detect_suite(file_path_from_test):
A CUDA file can carry multiple `register_cuda_ci(...)` calls — one per
pool it should run on — so this returns a *list* of dispatch dicts, one
per registration. Runner label, install script, timeout, and rdma_devices
are all resolved from scripts/ci/runner_configs.yml — the same single
source of truth that drives the main PR test pipeline.
per registration. Runner label, install script, timeout, grace_blackwell,
and rdma_devices are all resolved from scripts/ci/runner_configs.yml — the
same single source of truth that drives the main PR test pipeline.
Legacy nightly/weekly CUDA suites (single-string `suite=`) are dispatchable
too: each suite name is mapped to the matching runner_config via
@@ -806,7 +809,7 @@ def detect_suite(file_path_from_test):
`error` set.
Each dict has keys: suite, runner_label, install_script,
install_timeout, rdma_devices, is_cpu, error.
install_timeout, grace_blackwell, rdma_devices, is_cpu, error.
"""
full_path = f"test/{file_path_from_test}"
with open(full_path, "r") as f:
@@ -838,6 +841,7 @@ def detect_suite(file_path_from_test):
"runner_label": "ubuntu-latest",
"install_script": "",
"install_timeout": "",
"grace_blackwell": "0",
"rdma_devices": "",
"is_cpu": True,
"error": None,
@@ -912,6 +916,7 @@ def _resolve_test_spec(test_spec):
"runs_on": runner_label,
"install_script": "",
"install_timeout": "",
"grace_blackwell": "0",
"rdma_devices": "",
"error": None,
}
@@ -930,7 +935,8 @@ def _resolve_test_spec(test_spec):
print(
f"Resolved: file={resolved_path}, selector={test_selector}, "
f"suite={info['suite']}, mode={mode}, runs_on={info['runner_label']}, "
f"install={info['install_script']}, rdma={info['rdma_devices']}, "
f"install={info['install_script']}, grace_blackwell={info['grace_blackwell']}, "
f"rdma={info['rdma_devices']}, "
f"command='{test_command}'"
)
out.append(
@@ -941,6 +947,7 @@ def _resolve_test_spec(test_spec):
"runs_on": info["runner_label"],
"install_script": info["install_script"],
"install_timeout": info["install_timeout"],
"grace_blackwell": info["grace_blackwell"],
"rdma_devices": info["rdma_devices"],
"error": None,
}
@@ -952,7 +959,7 @@ def _dispatch_batch(gh_repo, pr, batch, token, reply_comment_id="", reply_marker
"""
Dispatch a single workflow run for a batch of resolved test specs that
share the same dispatch shape (mode + runs_on + install_script +
install_timeout + rdma_devices).
install_timeout + grace_blackwell + rdma_devices).
Returns a dict with keys: specs, success, test_commands, runner_label, run_url, error.
"""
@@ -961,6 +968,7 @@ def _dispatch_batch(gh_repo, pr, batch, token, reply_comment_id="", reply_marker
runs_on = batch[0]["runs_on"]
install_script = batch[0]["install_script"]
install_timeout = batch[0]["install_timeout"]
grace_blackwell = batch[0]["grace_blackwell"]
rdma_devices = batch[0]["rdma_devices"]
# Join multiple commands with newlines for the workflow to iterate over
@@ -993,6 +1001,7 @@ def _dispatch_batch(gh_repo, pr, batch, token, reply_comment_id="", reply_marker
"runs_on": runs_on or "",
"install_script": install_script,
"install_timeout": install_timeout or "20",
"grace_blackwell": grace_blackwell or "0",
"rdma_devices": rdma_devices,
"reply_comment_id": str(reply_comment_id) if reply_comment_id else "",
"reply_marker": reply_marker,
@@ -1100,7 +1109,7 @@ def handle_rerun_test(
"""
Handles the /rerun-test command. Resolves all test specs, groups them by
dispatch shape (mode + runs_on + install_script + install_timeout +
rdma_devices), and dispatches one workflow per group.
grace_blackwell + rdma_devices), and dispatches one workflow per group.
"""
if not skip_permission_check and not _check_rerun_test_permissions(
gh_repo, pr, comment, user_perms, "rerun-test"
@@ -1195,6 +1204,7 @@ def handle_rerun_test(
r["runs_on"],
r["install_script"],
r["install_timeout"],
r["grace_blackwell"],
r["rdma_devices"],
)
groups.setdefault(key, []).append(r)