hotfix: test_shard_no_recompile shouldn't rely on schedule order [pr] (#8928)

This commit is contained in:
qazal
2025-02-06 16:27:59 +02:00
committed by GitHub
parent 1249e8dd3b
commit 79fb5c6470
+1 -1
View File
@@ -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):