diff --git a/tests/python/unittest/test_auto_scheduler_layout_rewrite.py b/tests/python/unittest/test_auto_scheduler_layout_rewrite.py index e6f9a76fce62..9e4f1bf27735 100644 --- a/tests/python/unittest/test_auto_scheduler_layout_rewrite.py +++ b/tests/python/unittest/test_auto_scheduler_layout_rewrite.py @@ -19,6 +19,8 @@ import tempfile import numpy as np +import pytest + import tvm from tvm import topi from tvm import auto_scheduler, te @@ -46,6 +48,7 @@ def test_apply_steps_with_layout_rewrite(): assert bufs[1].shape[1] == 512 +@pytest.mark.skip("skip due to flaky") @tvm.testing.requires_llvm def test_correctness_layout_rewrite_rewrite_for_preTransformed(): N = 128 @@ -116,6 +119,7 @@ def test_correctness_layout_rewrite_rewrite_for_preTransformed(): del measure_ctx +@pytest.mark.skip("skip due to flaky") @tvm.testing.requires_llvm def test_correctness_layout_rewrite_insert_transform_stage(): N = 128