From 8504786b9ae7b1a078f9cf1ca520d995b6d47d73 Mon Sep 17 00:00:00 2001 From: Chris Sidebottom Date: Tue, 5 Oct 2021 13:23:14 +0000 Subject: [PATCH] [UnitTests][CMSISNN] Mark Binary Ops CMSIS NN tests as skipped https://github.com/apache/tvm/pull/9179 and https://github.com/apache/tvm/pull/9167 both landed at the same time which resulted in new tests without the skipping. --- tests/python/contrib/test_cmsisnn/test_binary_ops.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/python/contrib/test_cmsisnn/test_binary_ops.py b/tests/python/contrib/test_cmsisnn/test_binary_ops.py index 72e47e50b878..d785cfa199ae 100644 --- a/tests/python/contrib/test_cmsisnn/test_binary_ops.py +++ b/tests/python/contrib/test_cmsisnn/test_binary_ops.py @@ -22,6 +22,7 @@ import numpy as np import pytest +import tvm from tvm import relay from tvm.relay.op.contrib import cmsisnn @@ -61,6 +62,7 @@ def make_model( @skip_if_no_reference_system +@tvm.testing.requires_cmsisnn @pytest.mark.parametrize("op", [relay.qnn.op.mul, relay.qnn.op.add]) @pytest.mark.parametrize( [ @@ -129,6 +131,7 @@ def test_op_int8(op, input_0_scale, input_0_zero_point, input_1_scale, input_1_z ) +@tvm.testing.requires_cmsisnn @pytest.mark.parametrize("op", [relay.qnn.op.mul, relay.qnn.op.add]) @pytest.mark.parametrize(["input_dtype"], [["uint8"], ["int16"]]) def test_invalid_parameters(