fix test extra issue (#3159)

2026-06-24 02:14:17 +00:00 · 2024-01-17 11:58:08 -08:00 · 2024-01-17 11:58:08 -08:00 · ee83505fcc
commit ee83505fcc
parent 9cc2577a08
4 changed files with 2 additions and 0 deletions
--- a/test/test_multitensor.py
+++ b/test/test_multitensor.py
@ -24,12 +24,14 @@ class TestMultiTensor(unittest.TestCase):
    X.to_((d0, d1))
    for lb in X.lazydata.lbs:
      assert lb.shape == (256,)
+    (X + X).realize()

  def test_shard(self):
    X = Tensor.ones(256).contiguous().realize()
    X.shard_((d0, d1), 0)
    for lb in X.lazydata.lbs:
      assert lb.shape == (128,)
+    (X + X).realize()

  def test_shard_same_device(self):
    X = Tensor.ones(256).contiguous().realize()
--- a/test/testextra/init.py
+++ b/test/testextra/init.py
--- a/test/testextra/test_export_model.py
+++ b/test/testextra/test_export_model.py
--- a/test/testextra/test_lr_scheduler.py
+++ b/test/testextra/test_lr_scheduler.py