add stop server for simulator

apache · Jun 8, 2022 · 9ac0be9 · 9ac0be9
1 parent af22ddd
commit 9ac0be9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python/tvm/contrib/hexagon/build.py b/python/tvm/contrib/hexagon/build.py
@@ -595,7 +595,7 @@ def _start(self):
 
     def cleanup_directory(self):
         """Abstract method implementation. See description in HexagonLauncherRPC."""
-        shutil.rmtree(self._workspace)
+        pass
 
     def stop_server(self):
         """Abstract method implementation. See description in HexagonLauncherRPC."""

diff --git a/python/tvm/contrib/hexagon/pytest_plugin.py b/python/tvm/contrib/hexagon/pytest_plugin.py
@@ -213,6 +213,8 @@ def hexagon_launcher(
             launcher.start_server()
         yield launcher
     finally:
+        if android_serial_number == "simulator":
+            launcher.stop_server()
         launcher.cleanup_directory()