From aa331d124a5f20e5d5ec383b95c63c812f0892be Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 1 Jun 2021 11:26:36 +0100
Subject: [PATCH] [Test] catch streamlining fail in mnv1 end2end test
---
tests/end2end/test_end2end_mobilenet_v1.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/tests/end2end/test_end2end_mobilenet_v1.py b/tests/end2end/test_end2end_mobilenet_v1.py
index 79263a709..5bfe8e1ea 100644
--- a/tests/end2end/test_end2end_mobilenet_v1.py
+++ b/tests/end2end/test_end2end_mobilenet_v1.py
@@ -188,6 +188,10 @@ def test_end2end_mobilenet_streamline():
model = model.transform(GiveReadableTensorNames())
model = model.transform(InferDataTypes())
model.save(build_dir + "/end2end_mobilenet_streamlined.onnx")
+ assert (
+ len(model.get_nodes_by_op_type("Add")) == 1
+ ) # only final quantized bias Add op remains
+ assert len(model.get_nodes_by_op_type("Mul")) == 0 # no Mul ops remain
def test_end2end_mobilenet_lowering():
--
GitLab