未验证 提交 9e51e383 编写于 作者: L lilong12 提交者: GitHub

update, test=develop (#30047)

上级 d0056c32
...@@ -169,6 +169,7 @@ class TestDistBase(unittest.TestCase): ...@@ -169,6 +169,7 @@ class TestDistBase(unittest.TestCase):
path_id="0", path_id="0",
check_error_log=False, check_error_log=False,
need_envs={}): need_envs={}):
with_gloo = '0' if backend == "nccl" else '1'
required_envs = { required_envs = {
"FLAGS_fraction_of_gpu_memory_to_use": "0.15", "FLAGS_fraction_of_gpu_memory_to_use": "0.15",
"FLAGS_eager_delete_tensor_gb": "0.0", "FLAGS_eager_delete_tensor_gb": "0.0",
...@@ -178,7 +179,7 @@ class TestDistBase(unittest.TestCase): ...@@ -178,7 +179,7 @@ class TestDistBase(unittest.TestCase):
"LD_PRELOAD": os.getenv("LD_PRELOAD", ""), "LD_PRELOAD": os.getenv("LD_PRELOAD", ""),
"GLOG_v": "0", "GLOG_v": "0",
"NCCL_P2P_DISABLE": "1", "NCCL_P2P_DISABLE": "1",
"PADDLE_WITH_GLOO": "1", "PADDLE_WITH_GLOO": with_gloo,
"BACKEND": backend, "BACKEND": backend,
"PATH_ID": path_id "PATH_ID": path_id
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册