From 07bdedee1171ceb8d0157a072624ebde8f81c9a0 Mon Sep 17 00:00:00 2001 From: Johnson Zhang Date: Tue, 29 Apr 2025 16:56:53 +0800 Subject: [PATCH] [RPC] Fix Bug That Change Dict When Iterate The Keys --- python/tvm/rpc/proxy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/rpc/proxy.py b/python/tvm/rpc/proxy.py index b14b02eb56ad..293984f9ca9a 100644 --- a/python/tvm/rpc/proxy.py +++ b/python/tvm/rpc/proxy.py @@ -334,7 +334,7 @@ def _regenerate_server_keys(self, keys): keyset = set(self._server_pool.keys()) new_keys = [] # re-generate the server match key, so old information is invalidated. - for key in keys: + for key in tuple(keys): rpc_key, _ = base.split_random_key(key) handle = self._server_pool[key] del self._server_pool[key]