From e9cab6715235bc5c92318cae7b7cf3157e9df263 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Sun, 9 May 2021 23:34:30 +0100
Subject: [PATCH] [Test] catch bias export bug with xfail
---
tests/brevitas/test_brevitas_QConv2d.py | 2 ++
tests/brevitas/test_brevitas_qlinear.py | 2 ++
2 files changed, 4 insertions(+)
diff --git a/tests/brevitas/test_brevitas_QConv2d.py b/tests/brevitas/test_brevitas_QConv2d.py
index 5f124690d..21de8863d 100644
--- a/tests/brevitas/test_brevitas_QConv2d.py
+++ b/tests/brevitas/test_brevitas_QConv2d.py
@@ -50,6 +50,8 @@ 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 62ed358dc..e389bc8c1 100644
--- a/tests/brevitas/test_brevitas_qlinear.py
+++ b/tests/brevitas/test_brevitas_qlinear.py
@@ -48,6 +48,8 @@ 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