Fix nightly CI test for GLM-4.6 + B200 (#28749)
Co-authored-by: Brayden Zhong <brayden@radixark.ai>
This commit is contained in:
co-authored by
Brayden Zhong
parent
c01ad10d16
commit
d46afbf8b4
@@ -4327,7 +4327,7 @@ class ServerArgs:
|
||||
):
|
||||
self.quantization = quant_method
|
||||
if (
|
||||
self.quantization == "modelopt_fp4"
|
||||
self.quantization in {"modelopt_fp4", None}
|
||||
and self.moe_a2a_backend == "none"
|
||||
and self.moe_runner_backend == "auto"
|
||||
):
|
||||
|
||||
@@ -725,7 +725,7 @@ _LEGACY_SUITE_TO_RUNNER_CONFIG = {
|
||||
"nightly-perf-vlm-2-gpu": "2-gpu-large",
|
||||
"nightly-4-gpu": "4-gpu-h100",
|
||||
"nightly-4-gpu-b200": "4-gpu-b200",
|
||||
"nightly-8-gpu-common": "8-gpu-h200",
|
||||
"nightly-8-gpu-common": ["8-gpu-h200", "8-gpu-b200"],
|
||||
"nightly-8-gpu-h200": "8-gpu-h200",
|
||||
"nightly-kernel-8-gpu-h200": "8-gpu-h200",
|
||||
"nightly-precision-8-gpu-h200": "8-gpu-h200",
|
||||
@@ -830,10 +830,14 @@ def detect_suite(file_path_from_test):
|
||||
legacy_suites = _extract_legacy_suites(content)
|
||||
mappable = [s for s in legacy_suites if s in _LEGACY_SUITE_TO_RUNNER_CONFIG]
|
||||
if mappable:
|
||||
return [
|
||||
_resolve_runner_config(_LEGACY_SUITE_TO_RUNNER_CONFIG[s], full_path, s)
|
||||
for s in mappable
|
||||
]
|
||||
results = []
|
||||
for s in mappable:
|
||||
rcs = _LEGACY_SUITE_TO_RUNNER_CONFIG[s]
|
||||
if isinstance(rcs, str):
|
||||
rcs = [rcs]
|
||||
for rc in rcs:
|
||||
results.append(_resolve_runner_config(rc, full_path, s))
|
||||
return results
|
||||
|
||||
if re.search(r"^[^#\n]*register_cpu_ci\s*\(", content, re.MULTILINE):
|
||||
return [
|
||||
|
||||
Reference in New Issue
Block a user