From fef04adaa61317f2abcc32fcd6c2aa7d6058fdfd Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Wed, 22 Sep 2021 23:37:45 +0200
Subject: [PATCH] [Transform] exclude input fifos from RemoveShallowFIFOs
---
src/finn/transformation/fpgadataflow/set_fifo_depths.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/src/finn/transformation/fpgadataflow/set_fifo_depths.py b/src/finn/transformation/fpgadataflow/set_fifo_depths.py
index c06c34574..39eb04956 100644
--- a/src/finn/transformation/fpgadataflow/set_fifo_depths.py
+++ b/src/finn/transformation/fpgadataflow/set_fifo_depths.py
@@ -87,9 +87,14 @@ class RemoveShallowFIFOs(Transformation):
def apply(self, model):
shallow_fifos = []
for node in model.graph.node:
+ if len(node.input) > 0:
+ is_first_node = model.find_producer(node.input[0]) is None
+ else:
+ is_first_node = True
if (
node.op_type == "StreamingFIFO"
and getCustomOp(node).get_nodeattr("depth") <= self.shallow_threshold
+ and (not is_first_node)
):
# bypass shallow fifos
shallow_fifos.append(node)
--
GitLab