mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki 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
commit 83a8217cbf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tinygrad/ops.py
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]):