[RPC] Add fail-guard for termination time exception (#13651)

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.
apache · Dec 22, 2022 · 209845f · 209845f
1 parent 36d89a2
commit 209845f
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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():