diff --git a/tinygrad/uop/spec.py b/tinygrad/uop/spec.py index 71a1379768..0e16188cf1 100644 --- a/tinygrad/uop/spec.py +++ b/tinygrad/uop/spec.py @@ -114,7 +114,7 @@ program_spec = PatternMatcher([ (UPat(Ops.INDEX, src=(UPat(GroupOp.Defines).or_after(), UPat())), lambda: True), # LOAD (idx, alt_value) / LOAD(idx) / STORE(idx, val) - (UPat(Ops.LOAD, src=(UPat(Ops.INDEX, name="idx").or_casted(), UPat((Ops.VECTORIZE, Ops.VCONST, Ops.CONST)))), validate_index), + (UPat(Ops.LOAD, src=(UPat(Ops.INDEX, name="idx").or_casted(), UPat())), validate_index), (UPat(Ops.LOAD, src=(UPat(Ops.INDEX, name="idx").or_casted(), )), validate_index), (UPat(Ops.STORE, src=(UPat(Ops.INDEX, name="idx").or_casted(), UPat())), validate_index),