[RPC] Add fail-guard for termination time exception (apache#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.
fzi-peccia · Mar 27, 2023 · c9001a0 · c9001a0
1 parent 323ec09
commit c9001a0
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():