From e79f679283c853cbadf09bf41fd18bb9601a83ee Mon Sep 17 00:00:00 2001 From: qazal Date: Mon, 13 May 2024 11:03:58 +0300 Subject: [PATCH] what breaks --- tinygrad/engine/schedule.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py index 8dd1f94bea..ac8b5e30ca 100644 --- a/tinygrad/engine/schedule.py +++ b/tinygrad/engine/schedule.py @@ -129,10 +129,12 @@ def _recurse_lb(buf:LazyBuffer, realizes:Dict[LazyBuffer, None], allbufs:Dict[La if buf.base != buf: # realize all places where the buffer is expanded if prod(buf.base.st.shape) < prod(buf.st.shape): + """ if len(buf.st.views) == 1 and buf.st.views[-1].mask and all_int(buf.base.st.shape) and \ prod(buf.base.st.shape) >= prod([y-x for x,y in buf.st.views[-1].mask]): simple_pads.add(buf.base) - elif buf.base.op is UnaryOps.CAST and isinstance(buf.base.srcs[0].dtype, ImageDType) and isinstance(buf.base.arg[0], ImageDType): + """ + if buf.base.op is UnaryOps.CAST and isinstance(buf.base.srcs[0].dtype, ImageDType) and isinstance(buf.base.arg[0], ImageDType): pass # don't realize image to image casts. this is part of a larger problem else: realizes[buf.base] = None