diff --git a/tinygrad/schedule/rangeify.py b/tinygrad/schedule/rangeify.py index 4d3c63c448..f86299e69b 100644 --- a/tinygrad/schedule/rangeify.py +++ b/tinygrad/schedule/rangeify.py @@ -540,12 +540,9 @@ rangeify_codegen = PatternMatcher([ # TODO: this can be moved into codegen? (UPat(Ops.NOOP, name="x"), lambda x: x.src[0] if len(x.src) else None), - # fix broadcast dtype - (UPat(Ops.AFTER, name="a").broadcast(name="b"), lambda a,b: a.broadcast(len(b.src))), (UPat(Ops.DEFINE_LOCAL).f(Ops.AFTER, allow_any_len=True).broadcast(name="dg").f(Ops.INDEX, name="idx", allow_any_len=True), lambda dg,idx: None if isinstance(idx.dtype, PtrDType) else idx.replace(dtype=dg.dtype, arg=None).load(dtype=dg.dtype.base.scalar().vec(dg.dtype.vcount))), - (UPat(Ops.AFTER, name="a").gep(name="b"), lambda a,b: a.gep(b.arg)), (UPat(Ops.DEFINE_LOCAL).f(Ops.AFTER, allow_any_len=True).gep(name="dg").f(Ops.INDEX, name="idx", allow_any_len=True), lambda dg,idx: None if isinstance(idx.dtype, PtrDType) else idx.replace(dtype=dg.dtype, arg=None).load(dtype=dg.dtype.base.scalar().vec(dg.dtype.vcount))),