Trace P0 batch-level DP coordination

2026-07-13 17:45:08 +08:00
parent f1cd859eea
commit 7f4ae1708b
3 changed files with 82 additions and 16 deletions
--- a/scripts/collectivespec/run_p0_phase_trace.py
+++ b/scripts/collectivespec/run_p0_phase_trace.py
@@ -200,6 +200,7 @@ def main() -> int:
        "overlay": str(overlay),
        "run_id": args.run_id,
        "topology_claim": "inherited_from_base_spec",
+        "gpu_count": base.get("hardware", {}).get("gpu_count"),
        "request_count": args.request_count,
        "completion_tokens": args.completion_tokens,
        "policies": ["control_k3", "heterogeneous"],