mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

all cpu tests pass local

Browse source
This commit is contained in:
George Hotz 2026-05-26 19:28:46 -07:00
commit 2c8867c3e9
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
test/null/test_uop_graph.py
View file

@ -552,7 +552,9 @@ class TestUOpGraph(unittest.TestCase):
ld0 = uops[-2].src[-1] # -2 to skip SINK
# the gate and invalid value are deleted from ld1
self.assertEqual(ld0.src[0], smem.after(barrier).index(lidx+2, ptr=True))
new_barrier = ld0.src[0].src[0].src[1]
assert new_barrier.op is Ops.BARRIER
self.assertEqual(ld0.src[0], smem.after(new_barrier).slice(lidx+2))
def test_fold_gated_store(self):
glbl = UOp(Ops.PARAM, dtypes.int.ptr(), (), 0)
@ -564,7 +566,7 @@ class TestUOpGraph(unittest.TestCase):
uops = to_uops_list([st0, st1])
# only the second store happens
self.assertEqual(len(uops), 6) # +1 for SINK
self.assertEqual(uops[-2], glbl.index(idx1, ptr=True).store(val)) # -2 to skip SINK
self.assertEqual(uops[-2], glbl.slice(idx1).store(val)) # -2 to skip SINK
@unittest.skip("this is a uop type error")
def test_asserts_bad_gate(self):