add rpc flags to init

7 years ago · e6ddbfede7
parent 7357d8412e
commit e6ddbfede7
1 changed files with 3 additions and 0 deletions
--- a/python/paddle/fluid/init.py
+++ b/python/paddle/fluid/init.py
@ -121,6 +121,9 @@ def __bootstrap__():
        read_env_flags.append('rpc_server_profile_period')
        read_env_flags.append('rpc_server_profile_path')
        read_env_flags.append('enable_rpc_profiler')
+        read_env_flags.append('rpc_send_thread_num')
+        read_env_flags.append('rpc_get_thread_num')
+        read_env_flags.append('rpc_prefetch_thread_num')

    if core.is_compiled_with_cuda():
        read_env_flags += [