Fix for CI failure because of newly added passes

Change-Id: If52d9214aca6ec0d4bd7d52d89011c23cd79cd57
apache · Nov 5, 2021 · d17daf5 · d17daf5
1 parent 864fc25
commit d17daf5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/tests/python/contrib/test_cmsisnn/test_binary_ops.py b/tests/python/contrib/test_cmsisnn/test_binary_ops.py
@@ -131,6 +131,7 @@ def test_op_int8(op, input_0_scale, input_0_zero_point, input_1_scale, input_1_z
     )
 
 
+@skip_if_no_reference_system
 @tvm.testing.requires_cmsisnn
 @pytest.mark.parametrize("op", [relay.qnn.op.mul, relay.qnn.op.add])
 @pytest.mark.parametrize(["input_dtype"], [["uint8"], ["int16"]])

diff --git a/tests/python/contrib/test_cmsisnn/test_conv2d.py b/tests/python/contrib/test_cmsisnn/test_conv2d.py
@@ -253,6 +253,7 @@ def parameterize_for_invalid_model(test):
     )(test)
 
 
+@tvm.testing.requires_cmsisnn
 @parameterize_for_invalid_model
 def test_invalid_parameters(
     in_dtype,

diff --git a/tests/python/contrib/test_cmsisnn/test_extract_constants.py b/tests/python/contrib/test_cmsisnn/test_extract_constants.py
@@ -58,6 +58,7 @@ def set_external_func_attr(func, compiler, ext_symbol):
     return func
 
 
+@tvm.testing.requires_cmsisnn
 def test_external_function():
     y0_data = np.random.uniform(0, 1, (8, 8)).astype("float32")
     x0 = relay.var("x0", shape=(8, 8))
@@ -81,6 +82,7 @@ def test_external_function():
     relay.transform.InferType()(mod)
 
 
+@tvm.testing.requires_cmsisnn
 def test_nested_function():
     y1_data = np.random.uniform(0, 1, (8, 8)).astype("float32")
     x1 = relay.var("x1", shape=(8, 8))
@@ -109,6 +111,7 @@ def test_nested_function():
     relay.transform.InferType()(mod)
 
 
+@tvm.testing.requires_cmsisnn
 def test_multiple_functions():
     y20_data = np.random.uniform(0, 1, (8, 8)).astype("float32")
     x20 = relay.var("x20", shape=(8, 8))
@@ -143,6 +146,7 @@ def test_multiple_functions():
     relay.transform.InferType()(mod)
 
 
+@tvm.testing.requires_cmsisnn
 def test_main_function():
     x0 = relay.var("x0", shape=(8, 8))
     y0 = relay.var("y0", shape=(8, 8))