From b9df2db1c5f62b264f5896c22072c16387f5c7e9 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 2 Sep 2020 15:25:07 +0800
Subject: [PATCH] build(third_party): update megray commit id

GitOrigin-RevId: cc753f941b18272222ae25fb053793292e34bc08
---
 .../python/test/unit/test_distributed.py      |  1 -
 .../test/unit/distributed/test_functional.py  | 22 +++++++++----------
 .../test/unit/distributed/test_util.py        |  1 -
 3 files changed, 11 insertions(+), 13 deletions(-)

diff --git a/imperative/python/test/unit/test_distributed.py b/imperative/python/test/unit/test_distributed.py
index dd94ac05..dd5add7e 100644
--- a/imperative/python/test/unit/test_distributed.py
+++ b/imperative/python/test/unit/test_distributed.py
@@ -73,7 +73,6 @@ def test_init_process_group():
             assert p.exitcode == 0
 
     check("nccl")
-    check("ucx")
 
 
 @pytest.mark.skipif(
diff --git a/python_module/test/unit/distributed/test_functional.py b/python_module/test/unit/distributed/test_functional.py
index f4adcbd9..8bbaf7e0 100644
--- a/python_module/test/unit/distributed/test_functional.py
+++ b/python_module/test/unit/distributed/test_functional.py
@@ -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)
diff --git a/python_module/test/unit/distributed/test_util.py b/python_module/test/unit/distributed/test_util.py
index 30926b79..04b34244 100644
--- a/python_module/test/unit/distributed/test_util.py
+++ b/python_module/test/unit/distributed/test_util.py
@@ -102,7 +102,6 @@ def test_init_process_group():
         assert p0.exitcode == 0 and p1.exitcode == 0
 
     check("nccl")
-    check("ucx")
 
 
 @pytest.mark.skipif(
-- 
GitLab