提交 57546b4c 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: f540c5330275a3bfffa7235e586741da34bb50a1
上级 90e7cb00
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册