allow selecting UPat files in TRACK_MATCH_STATS [run_process_replay] (t…

…inygrad#6470)
grekiki2 · Sep 11, 2024 · e645a0e · e645a0e
1 parent 3cde150
commit e645a0e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tinygrad/ops.py b/tinygrad/ops.py
@@ -735,6 +735,7 @@ def print_match_stats():
     ret = [0,0,0.0,0.0]
     for k,v in sorted(list(match_stats.items()), key=lambda x: x[1][2]):
       loc_str = f"{k.location[0].split('/')[-1]}:{k.location[1]}"
+      if getenv("UPAT_FILE", loc_str) not in loc_str: continue
       print(f"{v[0]:6d} / {v[1]:7d} -- {v[3]*1000.:9.2f} / {v[2]*1000.:9.2f} ms -- {loc_str:15s}", k.printable())
       ret = [x+y for x,y in zip(ret, v)]
     print(f"{ret[0]:6d} / {ret[1]:7d} -- {ret[3]*1000.:9.2f} / {ret[2]*1000.:9.2f} ms -- TOTAL")