From 9ac0be9616f518cca23ad2e8ae3a57e4e7a28fec Mon Sep 17 00:00:00 2001 From: Mehrdad Hessar Date: Wed, 8 Jun 2022 16:56:39 +0000 Subject: [PATCH] add stop server for simulator --- python/tvm/contrib/hexagon/build.py | 2 +- python/tvm/contrib/hexagon/pytest_plugin.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/python/tvm/contrib/hexagon/build.py b/python/tvm/contrib/hexagon/build.py index 78d2261e156d9..5d912964a6668 100644 --- 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 index 7480e7f3ebe82..1841c654b934d 100644 --- 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()