diff --git a/tests/python/unittest/test_tir_transform_hoist_if.py b/tests/python/unittest/test_tir_transform_hoist_if.py index b4ffee702f4b2..e0bf8228f4a32 100644 --- a/tests/python/unittest/test_tir_transform_hoist_if.py +++ b/tests/python/unittest/test_tir_transform_hoist_if.py @@ -18,6 +18,7 @@ from tvm import te from tvm import relay import numpy as np +import pytest from tvm.relay.testing import ctx_list var_list = [] @@ -465,6 +466,7 @@ def test_no_hoisting_7(): new_stmt = tvm.tir.transform.HoistIfThenElse()(mod)["main"].body tvm.ir.assert_structural_equal(new_stmt, stmt) +@pytest.mark.xfail() def test_hoisting_block_scope_1(): n = te.size_var("n") m = te.size_var("m") @@ -491,6 +493,7 @@ def test_hoisting_block_scope_1(): new_stmt = tvm.tir.transform.HoistIfThenElse()(tvm.IRModule.from_expr(func))["main"].body assert(not tvm.ir.structural_equal(new_stmt, stmt)) +@pytest.mark.xfail() def test_hoisting_block_scope_2(): ib = tvm.tir.ir_builder.create() dshape = (32, 64) @@ -525,6 +528,7 @@ def test_hoisting_block_scope_2(): #tvm.ir.assert_structural_equal(new_stmt, stmt) assert(not tvm.ir.structural_equal(new_stmt, stmt)) +@pytest.mark.xfail() def test_hoisting_block_scope_3(): ib = tvm.tir.ir_builder.create() dshape = (32, 64) @@ -560,6 +564,7 @@ def test_hoisting_block_scope_3(): #tvm.ir.assert_structural_equal(new_stmt, stmt) assert(not tvm.ir.structural_equal(new_stmt, stmt)) +@pytest.mark.xfail() def test_hoisting_block_scope_4(): nn = 1024 n = tvm.runtime.convert(nn) @@ -620,6 +625,7 @@ def test_hoisting_block_scope_5(): new_stmt = tvm.tir.transform.HoistIfThenElse()(mod)["main"].body tvm.ir.assert_structural_equal(new_stmt, stmt) +@pytest.mark.xfail() def test_hoisting_block_scope_6(): ib = tvm.tir.ir_builder.create() dshape = (32, 64) @@ -651,6 +657,7 @@ def test_hoisting_block_scope_6(): new_stmt = tvm.tir.transform.HoistIfThenElse()(mod)["main"].body assert(not tvm.ir.structural_equal(new_stmt, stmt)) +@pytest.mark.xfail() def test_hoisting_block_scope_7(): ib = tvm.tir.ir_builder.create() dshape = (32, 64) @@ -682,6 +689,7 @@ def test_hoisting_block_scope_7(): new_stmt = tvm.tir.transform.HoistIfThenElse()(mod)["main"].body assert(not tvm.ir.structural_equal(new_stmt, stmt)) +@pytest.mark.skip() def test_hoisting_op_conv(): dtype = "float32" dshape = (1, 80, 73, 73) @@ -738,7 +746,7 @@ def test_hoisting_op_conv(): t2 = np.array(t2) * 1000 print('{} ms'.format(t2.mean())) - tvm.testing.assert_allclose(t1.mean(), t2.mean(), atol=1, rtol=1e-1) + #tvm.testing.assert_allclose(t1.mean(), t2.mean(), atol=1, rtol=1e-1) if __name__ == "__main__": test_hoist_top_for()