From 1475c994ee1bf685df8330dc4436199f0e6ae14b Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 26 May 2021 14:36:08 +0100
Subject: [PATCH] [Test] re-enable bias export test cases after Brevitas fix
---
tests/brevitas/test_brevitas_QConv2d.py | 2 --
tests/brevitas/test_brevitas_qlinear.py | 2 --
2 files changed, 4 deletions(-)
diff --git a/tests/brevitas/test_brevitas_QConv2d.py b/tests/brevitas/test_brevitas_QConv2d.py
index 21de8863d..5f124690d 100644
--- a/tests/brevitas/test_brevitas_QConv2d.py
+++ b/tests/brevitas/test_brevitas_QConv2d.py
@@ -50,8 +50,6 @@ export_onnx_path = "test_brevitas_conv.onnx"
@pytest.mark.parametrize("bias", [True, False])
@pytest.mark.parametrize("in_channels", [32])
def test_brevitas_QConv2d(dw, bias, in_channels):
- if bias:
- pytest.xfail("bias export bug")
ishape = (1, 32, 111, 111)
if dw is True:
groups = in_channels
diff --git a/tests/brevitas/test_brevitas_qlinear.py b/tests/brevitas/test_brevitas_qlinear.py
index e389bc8c1..62ed358dc 100644
--- a/tests/brevitas/test_brevitas_qlinear.py
+++ b/tests/brevitas/test_brevitas_qlinear.py
@@ -48,8 +48,6 @@ export_onnx_path = "test_brevitas_qlinear.onnx"
@pytest.mark.parametrize("w_bits", [4])
@pytest.mark.parametrize("i_dtype", [DataType.UINT4])
def test_brevitas_qlinear(bias, out_features, in_features, w_bits, i_dtype):
- if bias:
- pytest.xfail("bias export bug")
i_shape = (1, in_features)
w_shape = (out_features, in_features)
b_linear = QuantLinear(
--
GitLab