hotfix: simpler test_mnist_model

This commit is contained in:
George Hotz
2024-12-20 10:17:15 -08:00
parent 20eebbc61a
commit 074315ec08
2 changed files with 3 additions and 2 deletions

View File

@@ -696,6 +696,7 @@ class TestMultiTensor(unittest.TestCase):
t.shard_(devices, axis=0).realize()
assert all([lb is lb.base and lb.realized.base.size == 4 * 16 for lb in t.lazydata.lbs])
@unittest.skip("this is unreliable on OSX")
def test_clone(self):
t = Tensor.rand(16, 16).shard(devices_2, axis=None)
np.testing.assert_allclose(t.numpy(), t.clone().numpy())