forked from tinygrad/tinygrad
hotfix: test_shard_no_recompile shouldn't rely on schedule order [pr] (#8928)
This commit is contained in:
@@ -84,7 +84,7 @@ class TestMultiTensor(unittest.TestCase):
|
||||
for si, ei in zip(sched[:], lower_schedule(sched)):
|
||||
if isinstance(ei.prg, CompiledRunner): names.append(ei.prg.p.name)
|
||||
ei.run()
|
||||
assert names[-2] == names[-1], "function was relinearized"
|
||||
self.assertEqual(len(set(names)), 3), "function was relinearized"
|
||||
|
||||
@unittest.skip("this doesn't fold because shard_ calls contiguous on all lbs")
|
||||
def test_sharded_memory(self):
|
||||
|
||||
Reference in New Issue
Block a user