#0: Update pgm_dispatch_golden.json

Also add an optional --golden argument to compare_pgm_dispatch_perf_ci.py
tenstorrent · Jan 31, 2025 · 6c87f4c · 6c87f4c
1 parent d354e26
commit 6c87f4c
Show file tree

Hide file tree

Showing 2 changed files with 477 additions and 474 deletions.
diff --git a/tests/tt_metal/tt_metal/perf_microbenchmark/dispatch/compare_pgm_dispatch_perf_ci.py b/tests/tt_metal/tt_metal/perf_microbenchmark/dispatch/compare_pgm_dispatch_perf_ci.py
@@ -10,22 +10,25 @@
 import sys
 import pathlib
 
-golden = json.load(
-    open(
-        os.path.join(
-            pathlib.Path(__file__).parent.resolve(),
-            "pgm_dispatch_golden.json",
-        ),
-        "r",
-    )
-)
 
 THRESHOLD_PCT = 5
 
 parser = argparse.ArgumentParser(description="Compare benchmark JSON to golden")
 parser.add_argument("json", help="JSON file to compare", type=argparse.FileType("r"))
+parser.add_argument(
+    "-g",
+    "--golden",
+    help="Golden JSON file",
+    type=argparse.FileType("r"),
+    default=os.path.join(
+        pathlib.Path(__file__).parent.resolve(),
+        "pgm_dispatch_golden.json",
+    ),
+)
 args = parser.parse_args()
 
+golden = json.load(args.golden)
+
 result = json.load(args.json)
 
 golden_benchmarks = {}