rigsum.py 471 B

12345678
  1. """RigMark JSON summary: per-workload gate and median decode, cold prefill per depth."""
  2. import json, sys
  3. d = json.load(open(sys.argv[1]))
  4. for w, v in d["decode"].items():
  5. g = v.get("completion_gate", {})
  6. print(f"{w:12s} {v['decode_tokens_per_second']['median']:7.1f} tok/s gate {g.get('passed')}/{g.get('total')}")
  7. for depth, v in d["prefill"].items():
  8. print(f"prefill {depth}: cold {v['cold']['effective_prefill_tokens_per_second']['median']:,.0f} tok/s")