From ee3f90d18ed24ba4bbf05551919ec5bc7a921113 Mon Sep 17 00:00:00 2001 From: tqchen Date: Wed, 21 Dec 2022 11:56:22 -0500 Subject: [PATCH] [RPC] Add fail-guard for termination time exception This PR adds fail-guard to reduce error messages thrown during process termination time. Such error won't trigger test error but will bring extra message during exit time. --- python/tvm/rpc/proxy.py | 5 ++++- python/tvm/rpc/server.py | 5 ++++- python/tvm/rpc/server_ios_launcher.py | 5 ++++- 3 files changed, 12 insertions(+), 3 deletions(-) diff --git a/python/tvm/rpc/proxy.py b/python/tvm/rpc/proxy.py index 4c3144e7b5cd..d7027c88a4b5 100644 --- a/python/tvm/rpc/proxy.py +++ b/python/tvm/rpc/proxy.py @@ -643,7 +643,10 @@ def terminate(self): self.proc = None def __del__(self): - self.terminate() + try: + self.terminate() + except ImportError: + pass def websocket_proxy_server(url, key=""): diff --git a/python/tvm/rpc/server.py b/python/tvm/rpc/server.py index 7932e98aa20c..a144356f2e6d 100644 --- a/python/tvm/rpc/server.py +++ b/python/tvm/rpc/server.py @@ -499,4 +499,7 @@ def terminate(self): self.proc = None def __del__(self): - self.terminate() + try: + self.terminate() + except ImportError: + pass diff --git a/python/tvm/rpc/server_ios_launcher.py b/python/tvm/rpc/server_ios_launcher.py index 2e31586f6456..06ead1c3d165 100644 --- a/python/tvm/rpc/server_ios_launcher.py +++ b/python/tvm/rpc/server_ios_launcher.py @@ -335,7 +335,10 @@ def terminate(self): print(e) def __del__(self): - self.terminate() + try: + self.terminate() + except ImportError: + pass @staticmethod def is_compatible_environment():