This is an automated email from the ASF dual-hosted git repository.
tlopex 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 6627296c69 [S-TIR][Test] Mark meta_schedule tuning tests as skip
(#18863)
6627296c69 is described below
commit 6627296c69a963635a5ab6d7277c9c3a3e9ec88d
Author: Tianqi Chen <[email protected]>
AuthorDate: Mon Mar 2 12:49:18 2026 -0500
[S-TIR][Test] Mark meta_schedule tuning tests as skip (#18863)
This PR marks tuning related tests metaschedule tests as skip
---
.../python/s_tir/meta_schedule/test_meta_schedule_builder.py | 5 +++++
.../meta_schedule/test_meta_schedule_measure_callback.py | 2 ++
tests/python/s_tir/meta_schedule/test_meta_schedule_runner.py | 11 +++++++++++
.../test_meta_schedule_schedule_rule_apply_custom_rule.py | 1 +
4 files changed, 19 insertions(+)
diff --git a/tests/python/s_tir/meta_schedule/test_meta_schedule_builder.py
b/tests/python/s_tir/meta_schedule/test_meta_schedule_builder.py
index b6e2eed3e1..4f81c5d5de 100644
--- a/tests/python/s_tir/meta_schedule/test_meta_schedule_builder.py
+++ b/tests/python/s_tir/meta_schedule/test_meta_schedule_builder.py
@@ -110,6 +110,7 @@ def _check_build_results(builder_results:
list[BuilderResult]):
os.rmdir(os.path.dirname(artifact_path))
[email protected]("Tuning test - launches runner")
def test_meta_schedule_single_build():
"""Test meta schedule builder for a single build"""
mod = MatmulModule
@@ -120,6 +121,7 @@ def test_meta_schedule_single_build():
_check_build_results(builder_results)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_multiple_build():
"""Test meta schedule builder for multiple builds"""
builder = LocalBuilder()
@@ -158,6 +160,7 @@ def test_meta_schedule_error_handle_test_builder():
assert error_msg == "error"
[email protected]("Tuning test - launches runner")
def test_meta_schedule_error_handle_build_func():
"""Test the error handing during building"""
@@ -177,6 +180,7 @@ def test_meta_schedule_error_handle_build_func():
assert error_msg.startswith("LocalBuilder: An exception occurred")
[email protected]("Tuning test - launches runner")
def test_meta_schedule_error_handle_export_func():
"""Test the error handing during building"""
@@ -196,6 +200,7 @@ def test_meta_schedule_error_handle_export_func():
assert error_msg.startswith("LocalBuilder: An exception occurred")
[email protected]("Tuning test - launches runner")
def test_meta_schedule_error_handle_time_out():
"""Test the error handing time out during building"""
diff --git
a/tests/python/s_tir/meta_schedule/test_meta_schedule_measure_callback.py
b/tests/python/s_tir/meta_schedule/test_meta_schedule_measure_callback.py
index d2e5845e45..b6fa306ce0 100644
--- a/tests/python/s_tir/meta_schedule/test_meta_schedule_measure_callback.py
+++ b/tests/python/s_tir/meta_schedule/test_meta_schedule_measure_callback.py
@@ -123,6 +123,7 @@ def test_meta_schedule_measure_callback_as_string():
assert pattern.match(str(measure_callback))
[email protected]("Tuning test - launches runner")
def test_meta_schedule_measure_callback_update_cost_model_with_zero():
@ms.derived_object
class AllZeroRunnerFuture(ms.runner.PyRunnerFuture):
@@ -148,6 +149,7 @@ def
test_meta_schedule_measure_callback_update_cost_model_with_zero():
)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_measure_callback_update_cost_model_with_runtime_error():
@ms.derived_object
class EmptyRunnerFuture(ms.runner.PyRunnerFuture):
diff --git a/tests/python/s_tir/meta_schedule/test_meta_schedule_runner.py
b/tests/python/s_tir/meta_schedule/test_meta_schedule_runner.py
index 6ad493dde0..840e70fdb5 100644
--- a/tests/python/s_tir/meta_schedule/test_meta_schedule_runner.py
+++ b/tests/python/s_tir/meta_schedule/test_meta_schedule_runner.py
@@ -163,6 +163,7 @@ def _clean_build(artifact_path: str) -> None:
raise RuntimeError("Unable to find remove_build_dir function.")
[email protected]("Tuning test - launches runner")
def test_meta_schedule_rpc_single_run():
"""Test meta schedule rpc runner for a single run"""
# Build the module
@@ -209,6 +210,7 @@ def test_meta_schedule_rpc_single_run():
_clean_build(builder_result.artifact_path)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_local_single_run():
"""Test meta schedule local runner for a single run"""
# Build the module
@@ -247,6 +249,7 @@ def test_meta_schedule_local_single_run():
_clean_build(builder_result.artifact_path)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_rpc_multiple_runs():
"""Test meta schedule rpc runner for multiple runs"""
# Build the module
@@ -316,6 +319,7 @@ def test_meta_schedule_rpc_multiple_runs():
_clean_build(builder_result.artifact_path)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_local_multiple_runs():
"""Test meta schedule local runner for multiple runs"""
# Build the module
@@ -379,6 +383,7 @@ def test_meta_schedule_local_multiple_runs():
_clean_build(builder_result.artifact_path)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_py_runner():
"""Test meta schedule PyRunner"""
@@ -392,6 +397,7 @@ def test_meta_schedule_py_runner():
runner.run([])
[email protected]("Tuning test - launches runner")
@tvm.testing.skip_if_32bit(reason="skipping test for i386.")
def test_meta_schedule_rpc_runner_time_out():
"""Test meta schedule RPC Runner time out by using a super large
workload"""
@@ -439,6 +445,7 @@ def test_meta_schedule_rpc_runner_time_out():
assert runner_result.run_secs is None
[email protected]("Tuning test - launches runner")
def test_meta_schedule_local_runner_time_out():
"""Test meta schedule Local Runner time out"""
mod = MatmulModule
@@ -541,6 +548,7 @@ def test_meta_schedule_rpc_runner_exception():
assert runner_result.run_secs is None
[email protected]("Tuning test - launches runner")
def test_meta_schedule_local_runner_exception():
"""Test meta schedule Local Runner exception"""
mod = MatmulModule
@@ -592,6 +600,7 @@ def test_meta_schedule_local_runner_exception():
_clean_build(builder_result.artifact_path)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_runner_matmul_test():
"""Test meta schedule runner with add module"""
@@ -702,6 +711,7 @@ def test_meta_schedule_runner_matmul_test():
_clean_build(builder_result.artifact_path)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_runner_add_test():
"""Test meta schedule runner with add module"""
@@ -814,6 +824,7 @@ def test_meta_schedule_runner_add_test():
_clean_build(builder_result.artifact_path)
[email protected]("Tuning test - launches runner")
def test_meta_schedule_local_runner_add_test():
"""Test meta schedule local runner with add module"""
diff --git
a/tests/python/s_tir/meta_schedule/test_meta_schedule_schedule_rule_apply_custom_rule.py
b/tests/python/s_tir/meta_schedule/test_meta_schedule_schedule_rule_apply_custom_rule.py
index eee51ae93b..607327c0e5 100644
---
a/tests/python/s_tir/meta_schedule/test_meta_schedule_schedule_rule_apply_custom_rule.py
+++
b/tests/python/s_tir/meta_schedule/test_meta_schedule_schedule_rule_apply_custom_rule.py
@@ -47,6 +47,7 @@ def sch_fn(sch: tvm.s_tir.Schedule, block: tvm.tir.SBlock) ->
list[tvm.s_tir.Sch
raise ValueError("Intended for
s_tir.meta_schedule.cpu.test_apply_custom_rule")
[email protected]("Tuning test - launches runner")
def test_custom_rule():
with pytest.raises(ValueError) as e_info:
with tempfile.TemporaryDirectory() as tmpdir: