[tvm] branch main updated: [Test] Fix flaky LocalRunner test due to a small timeout (#9181)

[sy...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

syfeng 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 6a32ac5  [Test] Fix flaky LocalRunner test due to a small timeout (#9181)
6a32ac5 is described below

commit 6a32ac57a789cfc74f7553cc7bfe023e7e3b1086
Author: Junru Shao <ju...@gmail.com>
AuthorDate: Sat Oct 2 20:50:28 2021 -0700

    [Test] Fix flaky LocalRunner test due to a small timeout (#9181)
---
 python/tvm/meta_schedule/runner/local_runner.py    | 2 +-
 tests/python/unittest/test_meta_schedule_runner.py | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/python/tvm/meta_schedule/runner/local_runner.py b/python/tvm/meta_schedule/runner/local_runner.py
index f6bc2bc..9ba1b17 100644
--- a/python/tvm/meta_schedule/runner/local_runner.py
+++ b/python/tvm/meta_schedule/runner/local_runner.py
@@ -175,7 +175,7 @@ class LocalRunner(PyRunner):
 
     def __init__(
         self,
-        timeout_sec: float,
+        timeout_sec: float = 30,
         evaluator_config: Optional[EvaluatorConfig] = None,
         cooldown_sec: float = 0.0,
         alloc_repeat: int = 1,
diff --git a/tests/python/unittest/test_meta_schedule_runner.py b/tests/python/unittest/test_meta_schedule_runner.py
index c2c180a..8cd8709 100644
--- a/tests/python/unittest/test_meta_schedule_runner.py
+++ b/tests/python/unittest/test_meta_schedule_runner.py
@@ -507,7 +507,7 @@ def test_meta_schedule_rpc_runner_exception():
 
 
 def test_meta_schedule_local_runner_exception():
-    """Test meta schedule Local Runner time out"""
+    """Test meta schedule Local Runner exception"""
     mod = MatmulModule
     builder = LocalBuilder()
     (builder_result,) = builder.build([BuilderInput(mod, Target("llvm"))])
@@ -541,7 +541,6 @@ def test_meta_schedule_local_runner_exception():
     )
 
     runner = LocalRunner(
-        timeout_sec=1,
         evaluator_config=evaluator_config,
         initializer=initializer,
         f_alloc_argument="meta_schedule.runner.test_exception",