Skip to content

Commit d17daf5

Browse files
author
Ashutosh Parkhi
committed
Fix for CI failure because of newly added passes
Change-Id: If52d9214aca6ec0d4bd7d52d89011c23cd79cd57
1 parent 864fc25 commit d17daf5

File tree

3 files changed

+6
-0
lines changed

3 files changed

+6
-0
lines changed

tests/python/contrib/test_cmsisnn/test_binary_ops.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -131,6 +131,7 @@ def test_op_int8(op, input_0_scale, input_0_zero_point, input_1_scale, input_1_z
131131
)
132132

133133

134+
@skip_if_no_reference_system
134135
@tvm.testing.requires_cmsisnn
135136
@pytest.mark.parametrize("op", [relay.qnn.op.mul, relay.qnn.op.add])
136137
@pytest.mark.parametrize(["input_dtype"], [["uint8"], ["int16"]])

tests/python/contrib/test_cmsisnn/test_conv2d.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -253,6 +253,7 @@ def parameterize_for_invalid_model(test):
253253
)(test)
254254

255255

256+
@tvm.testing.requires_cmsisnn
256257
@parameterize_for_invalid_model
257258
def test_invalid_parameters(
258259
in_dtype,

tests/python/contrib/test_cmsisnn/test_extract_constants.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ def set_external_func_attr(func, compiler, ext_symbol):
5858
return func
5959

6060

61+
@tvm.testing.requires_cmsisnn
6162
def test_external_function():
6263
y0_data = np.random.uniform(0, 1, (8, 8)).astype("float32")
6364
x0 = relay.var("x0", shape=(8, 8))
@@ -81,6 +82,7 @@ def test_external_function():
8182
relay.transform.InferType()(mod)
8283

8384

85+
@tvm.testing.requires_cmsisnn
8486
def test_nested_function():
8587
y1_data = np.random.uniform(0, 1, (8, 8)).astype("float32")
8688
x1 = relay.var("x1", shape=(8, 8))
@@ -109,6 +111,7 @@ def test_nested_function():
109111
relay.transform.InferType()(mod)
110112

111113

114+
@tvm.testing.requires_cmsisnn
112115
def test_multiple_functions():
113116
y20_data = np.random.uniform(0, 1, (8, 8)).astype("float32")
114117
x20 = relay.var("x20", shape=(8, 8))
@@ -143,6 +146,7 @@ def test_multiple_functions():
143146
relay.transform.InferType()(mod)
144147

145148

149+
@tvm.testing.requires_cmsisnn
146150
def test_main_function():
147151
x0 = relay.var("x0", shape=(8, 8))
148152
y0 = relay.var("y0", shape=(8, 8))

0 commit comments

Comments
 (0)