From 68de3fcd46ffe434eae81e34f280bacaa737e84d Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Wed, 22 Sep 2021 10:22:01 +0200
Subject: [PATCH] [Build] add AbsorbConsecutiveTransposes to std lowering flow
---
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 f7a1fef06..37d35a6b4 100644
--- a/src/finn/builder/build_dataflow_steps.py
+++ b/src/finn/builder/build_dataflow_steps.py
@@ -221,6 +221,7 @@ def step_streamline(model: ModelWrapper, cfg: DataflowBuildConfig):
model = model.transform(MakeMaxPoolNHWC())
model = model.transform(absorb.AbsorbTransposeIntoMultiThreshold())
model = model.transform(MakeMaxPoolNHWC())
+ model = model.transform(absorb.AbsorbConsecutiveTransposes())
model = model.transform(ConvertBipolarMatMulToXnorPopcount())
model = model.transform(Streamline())
# absorb final add-mul nodes into TopK
--
GitLab