diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py index 41142b6960..30a708aab7 100644 --- a/tinygrad/engine/schedule.py +++ b/tinygrad/engine/schedule.py @@ -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