From 76606eb386446eff010e86143cfd3d204523fcad Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 5 May 2026 15:34:05 -0700 Subject: [PATCH] push --- tinygrad/uop/spec.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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),