Skip to content

Commit 55c2f6f

Browse files
committed
[Test] add MNv1 for quant conv lowering test
1 parent 8d1ee1d commit 55c2f6f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/transformation/test_conv_lowering.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@
4646
from qonnx.util.test import download_model, get_golden_in_and_output
4747

4848

49-
def test_conv_lowering_quant_weights():
50-
model_name = "FINN-CNV_W2A2"
49+
@pytest.mark.parametrize("model_name", ["FINN-CNV_W2A2", "MobileNetv1-w4a4"])
50+
def test_conv_lowering_quant_weights(model_name):
5151
model = download_model(model_name, return_modelwrapper=True, do_cleanup=True)
5252
model = model.transform(LowerConvsToMatMul())
5353
assert model.get_nodes_by_op_type("Conv") == []

0 commit comments

Comments
 (0)