diff --git a/tinygrad/uop/spec.py b/tinygrad/uop/spec.py index 31545e8395..c72babb661 100644 --- a/tinygrad/uop/spec.py +++ b/tinygrad/uop/spec.py @@ -291,7 +291,9 @@ full_spec = PatternMatcher([ # temp VECTORIZE/INDEX during rewrite have the wrong dtype (UPat(Ops.STACK), lambda: True), - (UPat(Ops.INDEX), lambda: True), + + # no more bool in index + (UPat(Ops.INDEX, name="idx"), lambda idx: not any([dtypes.is_bool(x.dtype) for x in idx.src[1:]])), # all loads/stores (UPat((Ops.LOAD, Ops.STORE)), lambda: True),