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
|
self.quantization = quant_method
|
||||||
if (
|
if (
|
||||||
self.quantization == "modelopt_fp4"
|
self.quantization in {"modelopt_fp4", None}
|
||||||
and self.moe_a2a_backend == "none"
|
and self.moe_a2a_backend == "none"
|
||||||
and self.moe_runner_backend == "auto"
|
and self.moe_runner_backend == "auto"
|
||||||
):
|
):
|
||||||
|
|||||||
@@ -725,7 +725,7 @@ _LEGACY_SUITE_TO_RUNNER_CONFIG = {
|
|||||||
"nightly-perf-vlm-2-gpu": "2-gpu-large",
|
"nightly-perf-vlm-2-gpu": "2-gpu-large",
|
||||||
"nightly-4-gpu": "4-gpu-h100",
|
"nightly-4-gpu": "4-gpu-h100",
|
||||||
"nightly-4-gpu-b200": "4-gpu-b200",
|
"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-8-gpu-h200": "8-gpu-h200",
|
||||||
"nightly-kernel-8-gpu-h200": "8-gpu-h200",
|
"nightly-kernel-8-gpu-h200": "8-gpu-h200",
|
||||||
"nightly-precision-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)
|
legacy_suites = _extract_legacy_suites(content)
|
||||||
mappable = [s for s in legacy_suites if s in _LEGACY_SUITE_TO_RUNNER_CONFIG]
|
mappable = [s for s in legacy_suites if s in _LEGACY_SUITE_TO_RUNNER_CONFIG]
|
||||||
if mappable:
|
if mappable:
|
||||||
return [
|
results = []
|
||||||
_resolve_runner_config(_LEGACY_SUITE_TO_RUNNER_CONFIG[s], full_path, s)
|
for s in mappable:
|
||||||
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):
|
if re.search(r"^[^#\n]*register_cpu_ci\s*\(", content, re.MULTILINE):
|
||||||
return [
|
return [
|
||||||
|
|||||||
Reference in New Issue
Block a user