From b6d08f247d86b4ae0e4d934896a9f409c50e4432 Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Wed, 31 Dec 2025 13:37:57 -0500
Subject: [PATCH] assert z3_xor input type (#13933)

---
 tinygrad/uop/validate.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tinygrad/uop/validate.py b/tinygrad/uop/validate.py
index 42a50d51e3..c73647cbf6 100644
--- a/tinygrad/uop/validate.py
+++ b/tinygrad/uop/validate.py
@@ -11,9 +11,8 @@ try:
   # IDIV is truncated division but z3 does euclidian division (floor if b>0 ceil otherwise); mod by power of two sometimes uses Ops.AND
   def z3_cdiv(a, b):return z3.If((a<0), z3.If(0<b, (a+(b-1))/b, (a-(b+1))/b), a/b)
   def z3_xor(a,b):
-    if isinstance(a, z3.BoolRef): return a^b
-    assert a==-1 or b==-1, "xor can only be used in indexing if one of the arguments is -1"
-    return -a-1 if b==-1 else -b-1
+    assert isinstance(a, z3.BoolRef), f"{type(a)=}, {a=}"
+    return a^b
   z3_alu: dict[Ops, Callable] = python_alu | {Ops.MOD: lambda a,b: a-z3_cdiv(a,b)*b, Ops.IDIV: z3_cdiv, Ops.SHR: lambda a,b: a/(2**b.as_long()),
     Ops.SHL: lambda a,b: a*(2**b.as_long()), Ops.AND: lambda a,b: a%(b+1) if isinstance(b, z3.ArithRef) else a&b, Ops.WHERE: z3.If, Ops.XOR: z3_xor,
     Ops.MAX: lambda a,b: z3.If(a<b, b, a),}