From 43378ad626460e11e7afd1cf8176c51fe592396b Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Mon, 11 Mar 2019 12:37:57 +0800 Subject: [PATCH] add flags to init --- paddle/fluid/operators/distributed/communicator.cc | 2 +- python/paddle/fluid/__init__.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/distributed/communicator.cc b/paddle/fluid/operators/distributed/communicator.cc index 06f7859f4f..6acb572de9 100644 --- a/paddle/fluid/operators/distributed/communicator.cc +++ b/paddle/fluid/operators/distributed/communicator.cc @@ -96,7 +96,7 @@ Communicator::Communicator(const RpcCtxMap &send_varname_to_ctx, VLOG(0) << "communicator_recv_wait_ms: " << FLAGS_communicator_recv_wait_ms; VLOG(0) << "communicator_thread_pool_size: " << FLAGS_communicator_thread_pool_size; - VLOG(0) << "communicator_max_merge_var_num" + VLOG(0) << "communicator_max_merge_var_num: " << FLAGS_communicator_max_merge_var_num; VLOG(0) << "communicator_fake_rpc: " << FLAGS_communicator_fake_rpc; send_scope_.reset(new Scope()); diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py index 8af5e1c509..c478c8ceee 100644 --- a/python/paddle/fluid/__init__.py +++ b/python/paddle/fluid/__init__.py @@ -143,6 +143,7 @@ def __bootstrap__(): read_env_flags.append('use_mkldnn') if core.is_compiled_with_dist(): + #env for rpc read_env_flags.append('rpc_deadline') read_env_flags.append('rpc_server_profile_path') read_env_flags.append('enable_rpc_profiler') @@ -150,10 +151,14 @@ def __bootstrap__(): read_env_flags.append('rpc_get_thread_num') read_env_flags.append('rpc_prefetch_thread_num') read_env_flags.append('rpc_disable_reuse_port') + + # env for communicator read_env_flags.append('communicator_independent_recv_thread') read_env_flags.append('communicator_send_queue_size') read_env_flags.append('communicator_recv_wait_ms') read_env_flags.append('communicator_thread_pool_size') + read_env_flags.append('communicator_max_merge_var_num') + read_env_flags.append('communicator_fake_rpc') if core.is_compiled_with_brpc(): read_env_flags.append('max_body_size') #set brpc max body size