push

2026-05-05 15:34:05 -07:00
parent e74bf441f0
commit 76606eb386
1 changed files with 3 additions and 1 deletions
@@ -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),