From 99d81dfcc6a87f6bef978655a6d403f1ee58aab1 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 5 Oct 2022 12:26:24 +0200
Subject: [PATCH] [Streamline] bugfix in AbsorbConsecutiveTransposes
without this, a single trailing Transpose at the end of the NN
gets eaten up
---
src/finn/transformation/streamline/absorb.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/finn/transformation/streamline/absorb.py b/src/finn/transformation/streamline/absorb.py
index a983e6775..50dcbaa0e 100644
--- a/src/finn/transformation/streamline/absorb.py
+++ b/src/finn/transformation/streamline/absorb.py
@@ -492,6 +492,8 @@ class AbsorbConsecutiveTransposes(Transformation):
if node.op_type == "Transpose":
next_nodes = model.find_consumers(node.output[0])
perms1 = list(get_by_name(node.attribute, "perm").ints)
+ if len(next_nodes) == 0:
+ continue
# check if all nodes after fork are opposite transposes
all_opposite_transposes = True
for next_node in next_nodes:
--
GitLab