From e425124041cd9390f51056e58ad377994773009b Mon Sep 17 00:00:00 2001 From: gongweibao Date: Tue, 22 Oct 2019 21:50:32 +0800 Subject: [PATCH] Wait pserver to complete initialization. (#20777) --- python/paddle/fluid/tests/unittests/test_dist_base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/test_dist_base.py b/python/paddle/fluid/tests/unittests/test_dist_base.py index 531342ce26..338b6e8b03 100644 --- a/python/paddle/fluid/tests/unittests/test_dist_base.py +++ b/python/paddle/fluid/tests/unittests/test_dist_base.py @@ -249,6 +249,9 @@ class TestDistRunnerBase(object): trainer_prog = fluid.default_main_program() print_to_err(type(self).__name__, "use main program done.") + # FIXME(gongwb):wait pserver initialization. + time.sleep(1) + if args.use_cuda: device_id = int(os.getenv("FLAGS_selected_gpus", "0")) place = fluid.CUDAPlace(device_id) @@ -879,7 +882,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,gen_nccl_id_op=10,nccl_helper=10,grpc_client=10,grpc_server=10" + "sparse_all_reduce_op_handle=10,gen_nccl_id_op=10,nccl_helper=10,grpc_client=10,grpc_server=10,request_handler_impl=10" required_envs["GLOG_logtostderr"] = "1" required_envs.update(need_envs) -- GitLab