From 1c5df0b8428c56ba9e207a40878f5556c21e493a Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Wed, 27 Oct 2021 23:17:50 +0200
Subject: [PATCH] [SetFolding] work with 1D SWG too
---
src/finn/transformation/fpgadataflow/set_folding.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/finn/transformation/fpgadataflow/set_folding.py b/src/finn/transformation/fpgadataflow/set_folding.py
index 64d7a0807..758b7efd6 100644
--- a/src/finn/transformation/fpgadataflow/set_folding.py
+++ b/src/finn/transformation/fpgadataflow/set_folding.py
@@ -150,7 +150,7 @@ class SetFolding(Transformation):
# also set the folding of the upsteam DW SWU
# which must be identical to this node
swu_node = model.find_producer(node.input[0])
- if swu_node.op_type == "ConvolutionInputGenerator":
+ if swu_node.op_type.startswith("ConvolutionInputGenerator"):
swu_node_inst = getCustomOp(swu_node)
pe = node_inst.get_nodeattr("PE")
swu_node_inst.set_nodeattr("SIMD", pe)
--
GitLab