Skip to content
Snippets Groups Projects
Commit 1475c994 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] re-enable bias export test cases after Brevitas fix

parent 1beaf3f7
No related branches found
No related tags found
No related merge requests found
...@@ -50,8 +50,6 @@ export_onnx_path = "test_brevitas_conv.onnx" ...@@ -50,8 +50,6 @@ export_onnx_path = "test_brevitas_conv.onnx"
@pytest.mark.parametrize("bias", [True, False]) @pytest.mark.parametrize("bias", [True, False])
@pytest.mark.parametrize("in_channels", [32]) @pytest.mark.parametrize("in_channels", [32])
def test_brevitas_QConv2d(dw, bias, in_channels): def test_brevitas_QConv2d(dw, bias, in_channels):
if bias:
pytest.xfail("bias export bug")
ishape = (1, 32, 111, 111) ishape = (1, 32, 111, 111)
if dw is True: if dw is True:
groups = in_channels groups = in_channels
......
...@@ -48,8 +48,6 @@ export_onnx_path = "test_brevitas_qlinear.onnx" ...@@ -48,8 +48,6 @@ export_onnx_path = "test_brevitas_qlinear.onnx"
@pytest.mark.parametrize("w_bits", [4]) @pytest.mark.parametrize("w_bits", [4])
@pytest.mark.parametrize("i_dtype", [DataType.UINT4]) @pytest.mark.parametrize("i_dtype", [DataType.UINT4])
def test_brevitas_qlinear(bias, out_features, in_features, w_bits, i_dtype): 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) i_shape = (1, in_features)
w_shape = (out_features, in_features) w_shape = (out_features, in_features)
b_linear = QuantLinear( b_linear = QuantLinear(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment