From 39f24c4773bb50d896ecec5b656e76f8be13e26f Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 7 Jul 2026 13:19:44 -0700 Subject: [PATCH] stack is fine --- test/backend/test_linearizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/backend/test_linearizer.py b/test/backend/test_linearizer.py index d0cf729223..5f1a5c1f78 100644 --- a/test/backend/test_linearizer.py +++ b/test/backend/test_linearizer.py @@ -251,9 +251,9 @@ class TestLinearizer(unittest.TestCase): for u in uops: if u.op is Ops.STORE and u.src[0].addrspace is AddrSpace.REG: if uops.index(u) < begin_range: - assert u.src[1].op is Ops.CONST + assert u.src[1].op is Ops.CONST or (u.src[1].op is Ops.STACK and all(x.op is Ops.CONST for x in u.src[1].src)) else: - assert u.src[1].op in GroupOp.ALU + assert u.src[1].op is Ops.CONST or (u.src[1].op is Ops.STACK and all(x.op in GroupOp.ALU for x in u.src[1].src)) assert begin_range < uops.index(u) < end_range # children of END are placed after ENDRANGE if any(x.op is Ops.END and x.src[1].op in GroupOp.ALU for x in u.src):