Skip to content

Commit 6a32ac5

Browse files
authored
[Test] Fix flaky LocalRunner test due to a small timeout (#9181)
1 parent 6b3fe95 commit 6a32ac5

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

python/tvm/meta_schedule/runner/local_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ def default_cleanup() -> None:
175175

176176
def __init__(
177177
self,
178-
timeout_sec: float,
178+
timeout_sec: float = 30,
179179
evaluator_config: Optional[EvaluatorConfig] = None,
180180
cooldown_sec: float = 0.0,
181181
alloc_repeat: int = 1,

tests/python/unittest/test_meta_schedule_runner.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -507,7 +507,7 @@ def exception_session_creator( # pylint: disable=unused-variable
507507

508508

509509
def test_meta_schedule_local_runner_exception():
510-
"""Test meta schedule Local Runner time out"""
510+
"""Test meta schedule Local Runner exception"""
511511
mod = MatmulModule
512512
builder = LocalBuilder()
513513
(builder_result,) = builder.build([BuilderInput(mod, Target("llvm"))])
@@ -541,7 +541,6 @@ def timeout_session_creator( # pylint: disable=unused-variable
541541
)
542542

543543
runner = LocalRunner(
544-
timeout_sec=1,
545544
evaluator_config=evaluator_config,
546545
initializer=initializer,
547546
f_alloc_argument="meta_schedule.runner.test_exception",

0 commit comments

Comments
 (0)