From 74fb405cc9915d4fc56bf3b3c35484edcaebcb18 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 3 Dec 2025 15:03:42 -0800 Subject: [PATCH] reenable the actual schedule cache --- tinygrad/engine/schedule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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