Browse Source

test(mge/distributed): fix test skip condition error

GitOrigin-RevId: f540c53302
release-1.2
Megvii Engine Team 4 years ago
parent
commit
57546b4c3d
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/python/test/unit/distributed/test_distributed.py

+ 1
- 1
imperative/python/test/unit/distributed/test_distributed.py View File

@@ -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


Loading…
Cancel
Save