This is an automated email from the ASF dual-hosted git repository.
leandron pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 7a00843 [RPC] Add Missing Command Line Option "through-proxy" of RPC
Server (#10188)
7a00843 is described below
commit 7a00843d146a5f975efc6e98e64e614e771a4885
Author: Qiang Zhang <[email protected]>
AuthorDate: Thu Feb 10 17:36:58 2022 +0800
[RPC] Add Missing Command Line Option "through-proxy" of RPC Server (#10188)
---
python/tvm/exec/rpc_server.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/python/tvm/exec/rpc_server.py b/python/tvm/exec/rpc_server.py
index ceab65c..4da88bc 100644
--- a/python/tvm/exec/rpc_server.py
+++ b/python/tvm/exec/rpc_server.py
@@ -42,6 +42,7 @@ def main(args):
args.host,
args.port,
args.port_end,
+ is_proxy=args.through_proxy,
key=args.key,
tracker_addr=tracker_addr,
load_library=args.load_library,
@@ -58,6 +59,15 @@ if __name__ == "__main__":
"--host", type=str, default="0.0.0.0", help="The host IP address the
tracker binds to"
)
parser.add_argument("--port", type=int, default=9090, help="The port of
the RPC")
+ parser.add_argument(
+ "--through-proxy",
+ dest="through_proxy",
+ action="store_true",
+ help=(
+ "Whether this server provide service through a proxy. If this is
true, the host and"
+ "port actually is the address of the proxy."
+ ),
+ )
parser.add_argument("--port-end", type=int, default=9199, help="The end
search port of the RPC")
parser.add_argument(
"--tracker",
