diff --git a/imperative/python/test/unit/distributed/test_distributed.py b/imperative/python/test/unit/distributed/test_distributed.py index d3e744a3ff3dd76b3d868c4ddc2069f0f82b54d8..2485d03e78e908d9016a193aaa6e05197f0a7f6a 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