Browse Source

build(third_party): update megray commit id

GitOrigin-RevId: cc753f941b
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
b9df2db1c5
3 changed files with 11 additions and 13 deletions
  1. +0
    -1
      imperative/python/test/unit/test_distributed.py
  2. +11
    -11
      python_module/test/unit/distributed/test_functional.py
  3. +0
    -1
      python_module/test/unit/distributed/test_util.py

+ 0
- 1
imperative/python/test/unit/test_distributed.py View File

@@ -73,7 +73,6 @@ def test_init_process_group():
assert p.exitcode == 0

check("nccl")
check("ucx")


@pytest.mark.skipif(


+ 11
- 11
python_module/test/unit/distributed/test_functional.py View File

@@ -64,7 +64,7 @@ def test_reduce_sum():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -106,7 +106,7 @@ def test_gather():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -144,7 +144,7 @@ def test_broadcast():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -186,7 +186,7 @@ def test_scatter():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (100, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -226,7 +226,7 @@ def test_all_to_all():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (100, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -265,7 +265,7 @@ def test_all_gather():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -308,7 +308,7 @@ def test_reduce_scatter_sum():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 4), (8, 10), (88, 44)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -347,7 +347,7 @@ def test_all_reduce_sum():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -386,7 +386,7 @@ def test_all_reduce_max():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -425,7 +425,7 @@ def test_all_reduce_min():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)


@@ -463,5 +463,5 @@ def test_bcast_param():
assert p0.exitcode == 0 and p1.exitcode == 0

for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]:
for backend in ["nccl"]:
check(shape, backend)

+ 0
- 1
python_module/test/unit/distributed/test_util.py View File

@@ -102,7 +102,6 @@ def test_init_process_group():
assert p0.exitcode == 0 and p1.exitcode == 0

check("nccl")
check("ucx")


@pytest.mark.skipif(


Loading…
Cancel
Save