sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity

hotfix: TRACK_MATCH_STATS=2 should not launch viz [pr] (#8543)

Browse Source
This commit is contained in:
qazal
2025-01-09 11:10:15 -05:00
committed by GitHub
parent 1efb1188d8
commit 83a8217cbf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/ops.py
+1 -1
View File
@@ -854,7 +854,7 @@ if TRACK_MATCH_STATS:
with open(fn:=temp("rewrites.pkl"), "wb") as f:
print(f"rewrote {len(tracked_ctxs)} graphs and matched {sum(len(r.matches) for x in tracked_ctxs for r in x)} times, saved to {fn}")
with Context(PICKLE_BUFFERS=0): pickle.dump((tracked_keys, tracked_ctxs), f)
launch_viz("VIZ", temp("rewrites.pkl"))
if getenv("VIZ"): launch_viz("VIZ", temp("rewrites.pkl"))
if getenv("PRINT_MATCH_STATS", 1):
ret = [0,0,0.0,0.0]
for k,v in sorted(list(match_stats.items()), key=lambda x: x[1][2]+x[1][3]):