From 48354ded387b553b994caedade63ebb08b3ebd30 Mon Sep 17 00:00:00 2001
From: Farshid Salemi Parizi <fparizi@octoml.ai>
Date: Thu, 11 Aug 2022 12:23:06 -0700
Subject: [PATCH] [Hexagon] Add skip option for RPC server initialization
 (#12368)

* add hardware parallelism in hexagon

* fix name

* lint issue

* better description
---
 python/tvm/contrib/hexagon/pytest_plugin.py | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/python/tvm/contrib/hexagon/pytest_plugin.py b/python/tvm/contrib/hexagon/pytest_plugin.py
index 1841c654b934..6b61b6f4ba55 100644
--- a/python/tvm/contrib/hexagon/pytest_plugin.py
+++ b/python/tvm/contrib/hexagon/pytest_plugin.py
@@ -155,7 +155,7 @@ def adb_server_socket() -> str:
 
 @pytest.fixture(scope="session")
 def hexagon_server_process(
-    request, android_serial_number, rpc_server_port_for_session, adb_server_socket
+    request, android_serial_number, rpc_server_port_for_session, adb_server_socket, skip_rpc
 ) -> HexagonLauncherRPC:
     """Initials and returns hexagon launcher if ANDROID_SERIAL_NUMBER is defined.
     This launcher is started only once per test session.
@@ -176,12 +176,13 @@ def hexagon_server_process(
             "adb_server_socket": adb_server_socket,
         }
         launcher = HexagonLauncher(serial_number=android_serial_number, rpc_info=rpc_info)
-
         try:
-            launcher.start_server()
+            if not skip_rpc:
+                launcher.start_server()
             yield launcher
         finally:
-            launcher.stop_server()
+            if not skip_rpc:
+                launcher.stop_server()
 
 
 @pytest.fixture
@@ -260,9 +261,21 @@ def aot_target(aot_host_target):
         assert False, "Incorrect AoT host target: {aot_host_target}. Options are [c, llvm]."
 
 
+@pytest.fixture(scope="session")
+def skip_rpc(request) -> bool:
+    return request.config.getoption("--skip-rpc")
+
+
 def pytest_addoption(parser):
     parser.addoption("--gtest_args", action="store", default="")
 
+    parser.addoption(
+        "--skip-rpc",
+        action="store_true",
+        default=False,
+        help="If set true, the RPC server initialization on Android would be skipped",
+    )
+
 
 def pytest_generate_tests(metafunc):
     option_value = metafunc.config.option.gtest_args