提交 b9df2db1 编写于 作者: M Megvii Engine Team

build(third_party): update megray commit id

GitOrigin-RevId: cc753f941b18272222ae25fb053793292e34bc08
上级 236c7ec8
...@@ -73,7 +73,6 @@ def test_init_process_group(): ...@@ -73,7 +73,6 @@ def test_init_process_group():
assert p.exitcode == 0 assert p.exitcode == 0
check("nccl") check("nccl")
check("ucx")
@pytest.mark.skipif( @pytest.mark.skipif(
......
...@@ -64,7 +64,7 @@ def test_reduce_sum(): ...@@ -64,7 +64,7 @@ def test_reduce_sum():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (99, 77)]: for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -106,7 +106,7 @@ def test_gather(): ...@@ -106,7 +106,7 @@ def test_gather():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (99, 77)]: for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -144,7 +144,7 @@ def test_broadcast(): ...@@ -144,7 +144,7 @@ def test_broadcast():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (99, 77)]: for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -186,7 +186,7 @@ def test_scatter(): ...@@ -186,7 +186,7 @@ def test_scatter():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (100, 77)]: for shape in [(2, 3), (8, 10), (100, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -226,7 +226,7 @@ def test_all_to_all(): ...@@ -226,7 +226,7 @@ def test_all_to_all():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (100, 77)]: for shape in [(2, 3), (8, 10), (100, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -265,7 +265,7 @@ def test_all_gather(): ...@@ -265,7 +265,7 @@ def test_all_gather():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (99, 77)]: for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -308,7 +308,7 @@ def test_reduce_scatter_sum(): ...@@ -308,7 +308,7 @@ def test_reduce_scatter_sum():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 4), (8, 10), (88, 44)]: for shape in [(2, 4), (8, 10), (88, 44)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -347,7 +347,7 @@ def test_all_reduce_sum(): ...@@ -347,7 +347,7 @@ def test_all_reduce_sum():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (99, 77)]: for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -386,7 +386,7 @@ def test_all_reduce_max(): ...@@ -386,7 +386,7 @@ def test_all_reduce_max():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (99, 77)]: for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -425,7 +425,7 @@ def test_all_reduce_min(): ...@@ -425,7 +425,7 @@ def test_all_reduce_min():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (99, 77)]: for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -463,5 +463,5 @@ def test_bcast_param(): ...@@ -463,5 +463,5 @@ def test_bcast_param():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
for shape in [(2, 3), (8, 10), (99, 77)]: for shape in [(2, 3), (8, 10), (99, 77)]:
for backend in ["nccl", "ucx"]: for backend in ["nccl"]:
check(shape, backend) check(shape, backend)
...@@ -102,7 +102,6 @@ def test_init_process_group(): ...@@ -102,7 +102,6 @@ def test_init_process_group():
assert p0.exitcode == 0 and p1.exitcode == 0 assert p0.exitcode == 0 and p1.exitcode == 0
check("nccl") check("nccl")
check("ucx")
@pytest.mark.skipif( @pytest.mark.skipif(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册