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"
@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
......
......@@ -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(
......
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