|
|
|
@ -848,6 +848,7 @@ class TestDistBase(unittest.TestCase):
|
|
|
|
|
"FLAGS_rpc_deadline": "30000", # 5sec to fail fast
|
|
|
|
|
"FLAGS_rpc_retry_bind_port": "50",
|
|
|
|
|
"FLAGS_cudnn_deterministic": "1",
|
|
|
|
|
"FLAGS_rpc_disable_reuse_port": "1",
|
|
|
|
|
"http_proxy": "",
|
|
|
|
|
"NCCL_P2P_DISABLE": "1",
|
|
|
|
|
"NCCL_SHM_DISABLE": "1"
|
|
|
|
@ -859,7 +860,7 @@ class TestDistBase(unittest.TestCase):
|
|
|
|
|
required_envs["GLOG_vmodule"] = \
|
|
|
|
|
"fused_all_reduce_op_handle=10,all_reduce_op_handle=10,alloc_continuous_space_op=10,fuse_all_reduce_op_pass=10," \
|
|
|
|
|
"alloc_continuous_space_for_grad_pass=10,fast_threaded_ssa_graph_executor=10,executor=10,operator=10," \
|
|
|
|
|
"sparse_all_reduce_op_handle=10"
|
|
|
|
|
"sparse_all_reduce_op_handle=10,gen_nccl_id_op=10"
|
|
|
|
|
required_envs["GLOG_logtostderr"] = "1"
|
|
|
|
|
|
|
|
|
|
local_losses \
|
|
|
|
|