Method test_allreduce_ring

test/backend/test_multitensor.py:227–230

(self)

      np.testing.assert_almost_equal(a.numpy(), b.numpy(), decimal=5)

  def test_allreduce_ring(self):
    with Context(RING=2):
      a,b = _test_allreduce(Tensor.rand(256, 256))
      np.testing.assert_almost_equal(a.numpy(), b.numpy(), decimal=5)

  def test_multiple_to_single_device(self):
    kernel_counts = {}