From e644d59f9f85f5cbe0841f4b9c5098259dfd3ff0 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 3 Dec 2025 14:07:04 -0800 Subject: [PATCH] oops, fix 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 fe09a31fcf..4a25cf3d27 100644 --- a/tinygrad/engine/schedule.py +++ b/tinygrad/engine/schedule.py @@ -153,7 +153,7 @@ def complete_create_schedule_with_vars(big_sink:UOp) -> tuple[dict[UOp, UOp], li # replace all UNIQUE buffers with LUNIQUE input_buffers: dict[UOp, UOp] = {} big_sink_cache = graph_rewrite(big_sink, pm_pre_sched_cache, ctx=input_buffers, name="rewrite for sched cache") - sched_cache_key = big_sink.key + sched_cache_key = big_sink_cache.key if (sc_ret:=schedule_cache.get(sched_cache_key, None)) is None: # verify Tensors match the spec (on big_sink, we only need to do this if cache misses)