From e26403588ef7899eeae9332a4f2a7e191451953c Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Mon, 10 May 2021 11:45:28 +0100
Subject: [PATCH] [Build] apply AbsorbSignBiasIntoMultiThreshold early on
---
src/finn/builder/build_dataflow_steps.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/finn/builder/build_dataflow_steps.py b/src/finn/builder/build_dataflow_steps.py
index 1c1861e52..4c9b076b6 100644
--- a/src/finn/builder/build_dataflow_steps.py
+++ b/src/finn/builder/build_dataflow_steps.py
@@ -158,6 +158,7 @@ def step_streamline(model: ModelWrapper, cfg: DataflowBuildConfig):
topologies.
"""
+ model = model.transform(absorb.AbsorbSignBiasIntoMultiThreshold())
model = model.transform(MoveScalarLinearPastInvariants())
model = model.transform(Streamline())
need_lowering = len(model.get_nodes_by_op_type("Conv")) > 0
--
GitLab