From cb6c72f57701d08e2095adaef32516898c19592a Mon Sep 17 00:00:00 2001 From: qazal Date: Sat, 7 Dec 2024 14:55:18 +0200 Subject: [PATCH] storing const is fine --- 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 0b9fddb92a..275628102f 100644 --- a/tinygrad/engine/schedule.py +++ b/tinygrad/engine/schedule.py @@ -389,7 +389,7 @@ def fold_img_cast(ctx:Dict[UOp, UOp], xb:UOp, view:UOp, b:UOp, to_cast:UOp, **kw return to_cast.view(unwrap(view.st)) def init_big_graph(ctx:ScheduleContext, sink:UOp) -> Optional[UOp]: - new_src = tuple(x.base for x in sink.src if is_scheduled(x.base) and uval(x.base).op is not Ops.CONST) + new_src = tuple(x.base for x in sink.src if is_scheduled(x.base) and x.base.src[1].op is not Ops.CONST) return None if new_src == sink.src else UOp(Ops.NOOP) if len(new_src) == 0 else UOp.sink(*new_src) do_realize = PatternMatcher([