This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new b738d31b87 [Hexagon][Metaschedule] Add timeout_sec arg to
get_hexagon_local_builder (#13828)
b738d31b87 is described below
commit b738d31b876c9ccd4a18b34400883db72340802b
Author: Alexey <[email protected]>
AuthorDate: Tue Jan 24 07:02:15 2023 +0300
[Hexagon][Metaschedule] Add timeout_sec arg to get_hexagon_local_builder
(#13828)
---
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 6e1541e498..92298c011d 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 get_hexagon_local_builder(
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(
