github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-14 17:38:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

reenable the actual schedule cache

Browse Source
This commit is contained in:
George Hotz
2025-12-03 15:03:42 -08:00
parent bf5de6ba5f
commit 74fb405cc9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/engine/schedule.py
View File

@@ -173,7 +173,7 @@ def complete_create_schedule_with_vars(big_sink:UOp) -> tuple[dict[UOp, UOp], li
# save in schedule cache
tensor_map_sink = UOp.sink(*flatten([(k,v) for k,v in tensor_map.items()]))
#schedule_cache[sched_cache_key] = (big_sink, tensor_map_sink)
schedule_cache[sched_cache_key] = (big_sink, tensor_map_sink)
else:
# schedule cache hit
del big_sink_cache