From 55d2055680e9082929c00f9001f9405b4991e9ba Mon Sep 17 00:00:00 2001 From: auphelia <jakobapk@web.de> Date: Thu, 9 Jul 2020 10:53:14 +0100 Subject: [PATCH] [Test] Delete lowering from brevitas QConv2 export test --- tests/brevitas/test_brevitas_QConv2d.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/tests/brevitas/test_brevitas_QConv2d.py b/tests/brevitas/test_brevitas_QConv2d.py index d904bc484..198f1e796 100644 --- a/tests/brevitas/test_brevitas_QConv2d.py +++ b/tests/brevitas/test_brevitas_QConv2d.py @@ -13,7 +13,6 @@ from finn.core.modelwrapper import ModelWrapper from finn.core.datatype import DataType import finn.core.onnx_exec as oxe from finn.transformation.infer_shapes import InferShapes -from finn.transformation.lower_convs_to_matmul import LowerConvsToMatMul from finn.util.basic import gen_finn_dt_tensor export_onnx_path = "test_brevitas_conv.onnx" @@ -75,8 +74,3 @@ def test_brevitas_QConv2d(dw, in_channels): assert np.isclose(produced, expected, atol=1e-3).all() os.remove(export_onnx_path) - - model = model.transform(LowerConvsToMatMul()) - odict = oxe.execute_onnx(model, idict, True) - produced_0 = odict[model.graph.output[0].name] - assert np.isclose(produced_0, expected, atol=1e-3).all() -- GitLab