提交 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():
assert p.exitcode == 0
check("nccl")
check("ucx")
@pytest.mark.skipif(
......
......@@ -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)
......@@ -102,7 +102,6 @@ def test_init_process_group():
assert p0.exitcode == 0 and p1.exitcode == 0
check("nccl")
check("ucx")
@pytest.mark.skipif(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册