From 56ee7cca453aec7938f87e24aec02c8ffe285b2c Mon Sep 17 00:00:00 2001
From: Frank <frank@dankware>
Date: Tue, 25 Aug 2026 17:12:51 -0700
Subject: [PATCH] Pass run number into ubatch benchmark parser

---
 scripts/bench-ubatch-rocmi4.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/scripts/bench-ubatch-rocmi4.sh b/scripts/bench-ubatch-rocmi4.sh
index 352050a..2acc49e 100755
--- a/scripts/bench-ubatch-rocmi4.sh
+++ b/scripts/bench-ubatch-rocmi4.sh
@@ -148,9 +148,9 @@ run_one() {
     local r
     for r in $(seq 1 "$runs"); do
         echo "--- run ${r}/${runs} ---"
-        python3 - "$ubatch" "$port" "$out_tokens" "$CTX" "$log" <<'PY'
+        python3 - "$ubatch" "$port" "$out_tokens" "$CTX" "$log" "$r" <<'PY'
 import sys, time, json, urllib.request
-ubatch, port, out_tokens, ctx, log = sys.argv[1], sys.argv[2], int(sys.argv[3]), int(sys.argv[4]), sys.argv[5]
+ubatch, port, out_tokens, ctx, log, run = sys.argv[1], sys.argv[2], int(sys.argv[3]), int(sys.argv[4]), sys.argv[5], sys.argv[6]
 
 # Flowing prose at ~75% of ctx so MTP acceptance is realistic.
 VAR = (
@@ -210,7 +210,7 @@ if evals and ct:
     if match:
         ms = float(match.group(1))
         g_tps = f"{ct / (ms / 1000):.2f}"
-print(f"== RESULT ubatch={ubatch} run={r} "
+print(f"== RESULT ubatch={ubatch} run={run} "
       f"PREFILL={grab(prefill)} GEN_TPS={g_tps} "
       f"pt={pt} ct={ct} wall={dt:.1f}s")
 PY