Skip to content

T
tic
Commit 657498a3 by Yizhi Liu Committed by Tianqi Chen
Options

fix rpc server proxy connect (#290)

parent 50ddb76b
Showing with 2 additions and 1 deletions
python/tvm/contrib/rpc.py
...@@ -90,12 +90,13 @@ def _listen_loop(sock): ...@@ -90,12 +90,13 @@ def _listen_loop(sock):
def _connect_proxy_loop(addr, key): def _connect_proxy_loop(addr, key):
key = "server:" + key
while True: while True:
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(addr) sock.connect(addr)
sock.sendall(struct.pack('@i', RPC_MAGIC)) sock.sendall(struct.pack('@i', RPC_MAGIC))
sock.sendall(struct.pack('@i', len(key))) sock.sendall(struct.pack('@i', len(key)))
sock.sendall(key) sock.sendall(key.encode("utf-8"))
magic = struct.unpack('@i', _recvall(sock, 4))[0] magic = struct.unpack('@i', _recvall(sock, 4))[0]
if magic == RPC_MAGIC + 1: if magic == RPC_MAGIC + 1:
raise RuntimeError("key: %s has already been used in proxy" % key) raise RuntimeError("key: %s has already been used in proxy" % key)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment