diff --git a/tests/python/unittest/test_meta_schedule_schedule_rule_auto_inline.py b/tests/python/unittest/test_meta_schedule_schedule_rule_auto_inline.py index e206fcc4502c..826054eb0b9b 100644 --- a/tests/python/unittest/test_meta_schedule_schedule_rule_auto_inline.py +++ b/tests/python/unittest/test_meta_schedule_schedule_rule_auto_inline.py @@ -16,6 +16,7 @@ # under the License. # pylint: disable=missing-module-docstring,missing-function-docstring,missing-class-docstring import tvm +import pytest from tvm.meta_schedule.space_generator.post_order_apply import PostOrderApply from tvm.meta_schedule.testing.schedule_rule import auto_inline from tvm.meta_schedule.tune_context import TuneContext @@ -270,6 +271,7 @@ def test_inline_consumer_chain(): tvm.ir.assert_structural_equal(lhs=space.mod, rhs=Conv2DBiasBnReLUInlined) +@pytest.mark.skip(reason="Flaky test") def test_inline_into_cache(): mod = MultiLevelTiledConv2D target = Target("cuda", host="llvm")