Method test_shard

test/backend/test_multitensor.py:52–57

(self)

    grad.realize()

  def test_shard(self):
    X = Tensor.ones(256).contiguous().realize()
    X.shard_(devices_2, 0)
    for lb in X.uop.src:
      assert lb.shape == (128,)
    (X + X).realize()

  @unittest.expectedFailure # TODO: fix
  def test_shard_empty(self):