From c75f61830839f15c6d6783e2839deac7fd9102cf Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 27 Apr 2021 14:57:45 +0100
Subject: [PATCH] [Test] enable bias on QConv2d export, currently failing
---
tests/brevitas/test_brevitas_QConv2d.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/tests/brevitas/test_brevitas_QConv2d.py b/tests/brevitas/test_brevitas_QConv2d.py
index 198f1e796..2307c2d7c 100644
--- a/tests/brevitas/test_brevitas_QConv2d.py
+++ b/tests/brevitas/test_brevitas_QConv2d.py
@@ -19,8 +19,9 @@ export_onnx_path = "test_brevitas_conv.onnx"
@pytest.mark.parametrize("dw", [False, True])
+@pytest.mark.parametrize("bias", [True, False])
@pytest.mark.parametrize("in_channels", [32])
-def test_brevitas_QConv2d(dw, in_channels):
+def test_brevitas_QConv2d(dw, bias, in_channels):
ishape = (1, 32, 111, 111)
if dw is True:
groups = in_channels
@@ -45,7 +46,7 @@ def test_brevitas_QConv2d(dw, in_channels):
kernel_size=kernel_size,
padding=padding,
stride=stride,
- bias=False,
+ bias=bias,
bias_quant_type=QuantType.FP,
compute_output_bit_width=False,
compute_output_scale=False,
--
GitLab