mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 22:48:25 -05:00
fix test extra issue (#3159)
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user