From 4c20012f107b5088c3dac5bb1749308c0314d5ba Mon Sep 17 00:00:00 2001 From: Eddie Yan Date: Thu, 14 Jun 2018 13:58:05 -0700 Subject: [PATCH] fix proxy registration with tracker --- python/tvm/contrib/rpc/proxy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/contrib/rpc/proxy.py b/python/tvm/contrib/rpc/proxy.py index b1168c1a843f..c7f66d68e492 100644 --- a/python/tvm/contrib/rpc/proxy.py +++ b/python/tvm/contrib/rpc/proxy.py @@ -333,7 +333,7 @@ def _update_tracker(self, period_update=False): rpc_key = key.split(":")[0] base.sendjson(self._tracker_conn, [TrackerCode.PUT, rpc_key, - (self._listen_port, key)]) + (self._listen_port, key), None]) assert base.recvjson(self._tracker_conn) == TrackerCode.SUCCESS if rpc_key not in self._key_set: self._key_set.add(rpc_key)