From 0fc686e9d0ed826046da7ab38f339148ea340afa Mon Sep 17 00:00:00 2001
From: Felix Jentzsch <45395194+fpjentzsch@users.noreply.github.com>
Date: Wed, 24 Feb 2021 13:39:02 +0100
Subject: [PATCH] Fix IODMA interface width selection (#287)

---
 src/finn/transformation/fpgadataflow/insert_iodma.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/finn/transformation/fpgadataflow/insert_iodma.py b/src/finn/transformation/fpgadataflow/insert_iodma.py
index 9a797bf02..671435475 100644
--- a/src/finn/transformation/fpgadataflow/insert_iodma.py
+++ b/src/finn/transformation/fpgadataflow/insert_iodma.py
@@ -138,7 +138,7 @@ class InsertIODMA(Transformation):
                 padded_instream_width = first_node_inst.get_instream_width_padded()
                 padded_instream_bytes = padded_instream_width // 8
                 # determine the feasible interface width
-                transfer_bits = padded_instream_width * np.prod(out_folded_shape[:-1])
+                transfer_bits = padded_instream_width * np.prod(in_folded_shape[:-1])
                 intfwidth = math.gcd(transfer_bits, self.max_intfwidth)
                 assert (
                     intfwidth % 8 == 0
-- 
GitLab