From eeae2f5768e23637fe9bf00cad809f0c8df10bdd Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 22 Oct 2025 19:40:13 +0800 Subject: [PATCH] after(barr) --- test/test_uops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_uops.py b/test/test_uops.py index 0cce5fd6c4..e0d2fb6cd3 100644 --- a/test/test_uops.py +++ b/test/test_uops.py @@ -342,7 +342,7 @@ class TestLocalAccess(unittest.TestCase): smem = uop(uops, Ops.DEFINE_LOCAL, dtypes.uint8.ptr(size=16, addrspace=AddrSpace.LOCAL), (), 'smem') st = uop(uops, Ops.STORE, dtypes.void, (smem.index(uop(uops, Ops.CONST, dtypes.int32, (), 0)), uop(uops, Ops.CONST, dtypes.uint8, (), 42))) barr = uop(uops, Ops.BARRIER, dtypes.void, (st,)) - sres = uop(uops, Ops.LOAD, dtypes.uint8, (smem.index(uop(uops, Ops.CONST, dtypes.int32, (), 0)), barr)) + sres = uop(uops, Ops.LOAD, dtypes.uint8, (smem.after(barr).index(uop(uops, Ops.CONST, dtypes.int32, (), 0)),)) self.assertEqual(_test_uops_result(dtypes.uint8, uops, sres), 42) # NOTE: webgpu specific, since only webgpu performs bitpacking