diff --git a/test/test_uops.py b/test/test_uops.py index dac9df01dd..332710d11a 100644 --- a/test/test_uops.py +++ b/test/test_uops.py @@ -151,7 +151,7 @@ class TestNonFloatUOps(TestUOps): value = -y+x uops = UOpGraph([UOp(UOps.STORE, None, (data0, idx, value))]) - with self.assertRaises(AssertionError): assert uops[-1].vin[2].arg is BinaryOps.SUB + assert uops[-1].vin[2].arg is BinaryOps.SUB class TestBoolUOps(TestUOps): def _test_uop_bool_fxn(self, op, fxn): diff --git a/tinygrad/codegen/uops.py b/tinygrad/codegen/uops.py index 430e1a7e70..b0f7eda1f1 100644 --- a/tinygrad/codegen/uops.py +++ b/tinygrad/codegen/uops.py @@ -181,7 +181,7 @@ constant_folder = PatternMatcher([ # -(-x) -> x (UPat(UOps.ALU, UnaryOps.NEG, (UPat(UOps.ALU, UnaryOps.NEG, (UPat(name="x"),)))), lambda x: x), # x+-y -> x-y - (UPat(UOps.ALU, BinaryOps.ADD, (UPat(name="x"), UPat(UOps.ALU, UnaryOps.NEG, name="my"))), lambda x, my: x-my.vin[0]), + (UPat(UOps.ALU, BinaryOps.ADD, [UPat(name="x"), UPat(UOps.ALU, UnaryOps.NEG, name="my")]), lambda x, my: x-my.vin[0]), # -1*x -> -x (UPat(UOps.ALU, BinaryOps.MUL, [UPat(name="x"), UPat(UOps.CONST, -1)]), lambda x: -x), # bool < False is always false, True < bool is always false