aryan5v commited on
Commit
cda326f
·
verified ·
1 Parent(s): 0ac7396

bench: include prompts in results bundle

Browse files
Files changed (1) hide show
  1. bench/run_matrix.py +7 -0
bench/run_matrix.py CHANGED
@@ -762,6 +762,13 @@ def orchestrate(args: argparse.Namespace) -> int:
762
  "quant_gemm": args.quant_gemm, "m5_sweep": args.m5_sweep}
763
  out.mkdir(parents=True, exist_ok=True)
764
  write_receipt(out, args, software, data, weights, settings)
 
 
 
 
 
 
 
765
  exit_code = 0
766
  lock = GpuLock(args.lock_dir, args.steal_stale_lock)
767
  # SIGINT is explicit: it is ignored by default when launched as a background job (nohup, agents).
 
762
  "quant_gemm": args.quant_gemm, "m5_sweep": args.m5_sweep}
763
  out.mkdir(parents=True, exist_ok=True)
764
  write_receipt(out, args, software, data, weights, settings)
765
+ # Ship the exact prompts/conditions with the results so the bundle is self-describing.
766
+ prompts_out = out / "prompts"
767
+ prompts_out.mkdir(exist_ok=True)
768
+ for name in ("index.json", "index.csv"):
769
+ source = Path(args.data_root) / "prompts" / name
770
+ if source.is_file():
771
+ shutil.copy2(source, prompts_out / name)
772
  exit_code = 0
773
  lock = GpuLock(args.lock_dir, args.steal_stale_lock)
774
  # SIGINT is explicit: it is ignored by default when launched as a background job (nohup, agents).