Browse Source

test(dist): remove redundant assert for reduce_sum and gather

GitOrigin-RevId: bc7393bb63
release-1.4
Megvii Engine Team 4 years ago
parent
commit
45a031fec3
1 changed files with 0 additions and 4 deletions
  1. +0
    -4
      imperative/python/test/unit/functional/test_functional_distributed.py

+ 0
- 4
imperative/python/test/unit/functional/test_functional_distributed.py View File

@@ -44,8 +44,6 @@ def test_reduce_sum(shape):
output = reduce_sum(inp)
if rank == 0:
assert np.allclose(output.numpy(), expect[rank])
else:
assert np.allclose(output.numpy(), 0)

x = np.random.random_sample(shape).astype("float32")
y = np.random.random_sample(shape).astype("float32")
@@ -179,8 +177,6 @@ def test_gather(shape):
output = gather(inp)
if rank == 0:
assert np.allclose(output.numpy(), expect[rank])
else:
assert np.allclose(output.numpy(), 0)

x = np.random.random_sample(shape).astype("float32")
y = np.random.random_sample(shape).astype("float32")


Loading…
Cancel
Save