Skip to content

T
tic
Commit d3c92aeb by eqy Committed by Tianqi Chen
Options

fix proxy registration with tracker (#1283)

parent 943dfa26
Showing with 1 additions and 1 deletions
python/tvm/contrib/rpc/proxy.py
...@@ -333,7 +333,7 @@ class ProxyServerHandler(object): ...@@ -333,7 +333,7 @@ class ProxyServerHandler(object):
rpc_key = key.split(":")[0] rpc_key = key.split(":")[0]
base.sendjson(self._tracker_conn, base.sendjson(self._tracker_conn,
[TrackerCode.PUT, rpc_key, [TrackerCode.PUT, rpc_key,
(self._listen_port, key)]) (self._listen_port, key), None])
assert base.recvjson(self._tracker_conn) == TrackerCode.SUCCESS assert base.recvjson(self._tracker_conn) == TrackerCode.SUCCESS
if rpc_key not in self._key_set: if rpc_key not in self._key_set:
self._key_set.add(rpc_key) self._key_set.add(rpc_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