failed test case for multi rand_like [pr] (#8740)

new multi broke multi device dropout
2025-01-24 13:56:51 -05:00
parent dc10187fc0
commit e0e176efbc
1 changed files with 6 additions and 0 deletions
@@ -651,6 +651,12 @@ class TestMultiTensor(unittest.TestCase):
    self.assertEqual(t.dtype, t2.dtype)
    self.assertEqual(t.lazydata.axis, t2.lazydata.axis)

+  def test_rand_like_from_alu(self):
+    a = Tensor.ones(4, 4).shard(devices_2, axis=0)
+    # TODO: fix this, which will also fix multi device dropout
+    with self.assertRaises(AssertionError):
+      (a + a).rand_like()
+
  @unittest.skip("no longer supports uneven shard")
  def test_rand_like_uneven_shard(self):
    t = Tensor.empty((4, 42, 15)).shard(devices_3, axis=1)