Skip to content

Commit a92093c

Browse files
committed
[Test] add (failing) quant weight conv testcase for lowering
1 parent 8f66615 commit a92093c

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/transformation/test_conv_lowering.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,14 @@
4343
from qonnx.transformation.infer_shapes import InferShapes
4444
from qonnx.transformation.lower_convs_to_matmul import LowerConvsToMatMul
4545
from qonnx.util.basic import gen_finn_dt_tensor, qonnx_make_model
46+
from qonnx.util.test import download_model
47+
48+
49+
def test_conv_lowering_quant_weights():
50+
model_name = "FINN-CNV_W2A2"
51+
model = download_model(model_name, return_modelwrapper=True, do_cleanup=True)
52+
model = model.transform(LowerConvsToMatMul())
53+
assert model.get_nodes_by_op_type("Conv") == []
4654

4755

4856
def test_conv_lowering_convmnist():

0 commit comments

Comments
 (0)