Skip to content

Commit

Permalink
[Test] add (failing) quant weight conv testcase for lowering
Browse files Browse the repository at this point in the history
  • Loading branch information
maltanar committed Aug 16, 2024
1 parent 8f66615 commit a92093c
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions tests/transformation/test_conv_lowering.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,14 @@
from qonnx.transformation.infer_shapes import InferShapes
from qonnx.transformation.lower_convs_to_matmul import LowerConvsToMatMul
from qonnx.util.basic import gen_finn_dt_tensor, qonnx_make_model
from qonnx.util.test import download_model


def test_conv_lowering_quant_weights():
model_name = "FINN-CNV_W2A2"
model = download_model(model_name, return_modelwrapper=True, do_cleanup=True)
model = model.transform(LowerConvsToMatMul())
assert model.get_nodes_by_op_type("Conv") == []


def test_conv_lowering_convmnist():
Expand Down

0 comments on commit a92093c

Please sign in to comment.