From fc4563e2f227b814080bcce743722ea43440f8f9 Mon Sep 17 00:00:00 2001 From: Alexey Voronov Date: Mon, 23 Jan 2023 15:10:20 +0000 Subject: [PATCH] [Hexagon][Metaschedule] Add timeout_sec arg to get_hexagon_local_builder --- python/tvm/contrib/hexagon/meta_schedule.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/python/tvm/contrib/hexagon/meta_schedule.py b/python/tvm/contrib/hexagon/meta_schedule.py index 6e1541e498a9..92298c011d4a 100644 --- a/python/tvm/contrib/hexagon/meta_schedule.py +++ b/python/tvm/contrib/hexagon/meta_schedule.py @@ -129,7 +129,9 @@ def _worker_func(hexagon_launcher, evaluator_config, alloc_repeat, artifact_path def get_hexagon_local_builder( - pass_context: tvm.transform.PassContext = None, max_workers: Optional[int] = None + pass_context: tvm.transform.PassContext = None, + max_workers: Optional[int] = None, + timeout_sec: float = 30.0, ): """Return Hexagon-compatible Builder for meta schedule.""" @@ -146,10 +148,13 @@ def default_build_with_context( if pass_context is not None: return LocalBuilder( - f_build=default_build_with_context, f_export=export_func, max_workers=max_workers + f_build=default_build_with_context, + f_export=export_func, + max_workers=max_workers, + timeout_sec=timeout_sec, ) else: - return LocalBuilder(f_export=export_func, max_workers=max_workers) + return LocalBuilder(f_export=export_func, max_workers=max_workers, timeout_sec=timeout_sec) def get_hexagon_rpc_runner(