Skip to content

Commit

Permalink
[UnitTests][CMSISNN] Mark Binary Ops CMSIS NN tests as skipped (apach…
Browse files Browse the repository at this point in the history
…e#9200)

apache#9179 and apache#9167 both landed at the same time which resulted in new tests without the skipping.
  • Loading branch information
Mousius authored and ylc committed Jan 13, 2022
1 parent ac6ad24 commit 458022f
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tests/python/contrib/test_cmsisnn/test_binary_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
import numpy as np
import pytest

import tvm
from tvm import relay
from tvm.relay.op.contrib import cmsisnn

Expand Down Expand Up @@ -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(
[
Expand Down Expand Up @@ -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(
Expand Down

0 comments on commit 458022f

Please sign in to comment.