From 30f0e95fbda6ae0de879610bd1d34e579e75d4e1 Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Fri, 29 Nov 2024 04:03:55 -0500 Subject: [PATCH] don't lru_cache is_scheduled [pr] (#7953) --- tinygrad/engine/schedule.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py index 76cb9615b8..1bb662fa7c 100644 --- a/tinygrad/engine/schedule.py +++ b/tinygrad/engine/schedule.py @@ -47,7 +47,6 @@ class ScheduleContext: ops_metadata: Dict[UOp, Metadata] = field(default_factory=dict) # this maps fused ops to Metadata children: DefaultDict[UOp, Dict[UOp, None]] = field(default_factory=lambda: defaultdict(dict)) -@functools.lru_cache(None) def is_scheduled(u:UOp) -> bool: return u.op is Ops.VIEW and len(u.src) == 2 def to_uop(buf:LazyBuffer, ctx:ScheduleContext, buffers:Dict[UOp, Buffer], cache:Dict[LazyBuffer, UOp]) -> UOp: