[tvm] branch main updated: [RPC] Add Missing Option "port_end" to RPC Proxy (#15116)

[syfeng]

This is an automated email from the ASF dual-hosted git repository.

syfeng 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 95c171458e [RPC] Add Missing Option "port_end" to RPC Proxy (#15116)
95c171458e is described below

commit 95c171458ef2651c88447480dc026e98bb05132b
Author: Qiang Zhang <johnson9...@163.com>
AuthorDate: Sat Jun 17 15:43:53 2023 +0800

    [RPC] Add Missing Option "port_end" to RPC Proxy (#15116)
---
 python/tvm/exec/rpc_proxy.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/python/tvm/exec/rpc_proxy.py b/python/tvm/exec/rpc_proxy.py
index 7eae4fe174..74c5770fca 100644
--- a/python/tvm/exec/rpc_proxy.py
+++ b/python/tvm/exec/rpc_proxy.py
@@ -57,13 +57,20 @@ def main(args):
         prox = Proxy(
             args.host,
             port=args.port,
+            port_end=args.port_end,
             web_port=args.web_port,
             index_page=index,
             resource_files=js_files,
             tracker_addr=tracker_addr,
         )
     else:
-        prox = Proxy(args.host, port=args.port, web_port=args.web_port, 
tracker_addr=tracker_addr)
+        prox = Proxy(
+            args.host,
+            port=args.port,
+            port_end=args.port_end,
+            web_port=args.web_port,
+            tracker_addr=tracker_addr,
+        )
     prox.proc.join()
 
 
@@ -71,6 +78,7 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument("--host", type=str, default="127.0.0.1", help="the 
hostname of the server")
     parser.add_argument("--port", type=int, default=9090, help="The port of 
the RPC")
+    parser.add_argument("--port-end", type=int, default=9199, help="The end 
search port of the RPC")
     parser.add_argument(
         "--web-port", type=int, default=8888, help="The port of the 
http/websocket server"
     )