From 8cec11289b2fe12363d4c8af28297e17939ef1ca Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 24 Aug 2020 18:17:16 +0200
Subject: [PATCH] [Test] remove 1-bit Brevitas QuantReLU testcases, unsupported
---
tests/brevitas/test_brevitas_relu_act_export.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/brevitas/test_brevitas_relu_act_export.py b/tests/brevitas/test_brevitas_relu_act_export.py
index 77974dacb..fa114585d 100644
--- a/tests/brevitas/test_brevitas_relu_act_export.py
+++ b/tests/brevitas/test_brevitas_relu_act_export.py
@@ -15,7 +15,7 @@ from finn.transformation.infer_shapes import InferShapes
export_onnx_path = "test_brevitas_relu_act_export.onnx"
-@pytest.mark.parametrize("abits", [1, 2, 4, 8])
+@pytest.mark.parametrize("abits", [2, 4, 8])
@pytest.mark.parametrize("max_val", [1.0, 1.5, 1 - 2 ** (-7)])
@pytest.mark.parametrize(
"scaling_impl_type", [ScalingImplType.CONST, ScalingImplType.PARAMETER]
@@ -70,7 +70,7 @@ scaling_impl.learned_value": torch.tensor(
os.remove(export_onnx_path)
-@pytest.mark.parametrize("abits", [1, 2, 4, 8])
+@pytest.mark.parametrize("abits", [2, 4, 8])
@pytest.mark.parametrize("max_val", [1.0, 1.5, 1 - 2 ** (-7)])
@pytest.mark.parametrize("scaling_per_channel", [True, False])
def test_brevitas_act_export_relu_imagenet(abits, max_val, scaling_per_channel):
--
GitLab