From 57546b4c3d4ee44482091fdac0059937bd2ed968 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 4 Dec 2020 12:25:08 +0800
Subject: [PATCH] test(mge/distributed): fix test skip condition error

GitOrigin-RevId: f540c5330275a3bfffa7235e586741da34bb50a1
---
 imperative/python/test/unit/distributed/test_distributed.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/imperative/python/test/unit/distributed/test_distributed.py b/imperative/python/test/unit/distributed/test_distributed.py
index d3e744a3..2485d03e 100644
--- a/imperative/python/test/unit/distributed/test_distributed.py
+++ b/imperative/python/test/unit/distributed/test_distributed.py
@@ -87,7 +87,7 @@ def test_init_process_group():
 @pytest.mark.skipif(
     platform.system() == "Windows", reason="windows disable MGB_ENABLE_OPR_MM"
 )
-@pytest.mark.skipif(get_device_count_by_fork("gpu") < 2, reason="need more gpu device")
+@pytest.mark.skipif(get_device_count_by_fork("gpu") < 3, reason="need more gpu device")
 @pytest.mark.isolated_distributed
 def test_new_group():
     world_size = 3