diff --git a/python/tvm/meta_schedule/runner/local_runner.py b/python/tvm/meta_schedule/runner/local_runner.py index f6bc2bcce55d..9ba1b1724161 100644 --- a/python/tvm/meta_schedule/runner/local_runner.py +++ b/python/tvm/meta_schedule/runner/local_runner.py @@ -175,7 +175,7 @@ def default_cleanup() -> None: 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 c2c180a201ac..8cd870931767 100644 --- a/tests/python/unittest/test_meta_schedule_runner.py +++ b/tests/python/unittest/test_meta_schedule_runner.py @@ -507,7 +507,7 @@ def exception_session_creator( # pylint: disable=unused-variable 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 timeout_session_creator( # pylint: disable=unused-variable ) runner = LocalRunner( - timeout_sec=1, evaluator_config=evaluator_config, initializer=initializer, f_alloc_argument="meta_schedule.runner.test_exception",