From 0242e6fe76490ce398f31d411a9386ae65294daf Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 12 Aug 2022 17:25:05 +0200
Subject: [PATCH] [FIFO] handle weight reps correctly for decoupled mode
---
.../transformation/fpgadataflow/derive_characteristic.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/src/finn/transformation/fpgadataflow/derive_characteristic.py b/src/finn/transformation/fpgadataflow/derive_characteristic.py
index fcec75024..82512b3f1 100644
--- a/src/finn/transformation/fpgadataflow/derive_characteristic.py
+++ b/src/finn/transformation/fpgadataflow/derive_characteristic.py
@@ -118,7 +118,10 @@ class DeriveCharacteristic(NodeLocalTransformation):
n_weight_inps = inst.calc_tmem()
else:
n_weight_inps = inst.calc_wmem()
- io_dict["inputs"]["weights"] = [0 for i in range(n_weight_inps)]
+ num_w_reps = np.prod(inst.get_nodeattr("numInputVectors"))
+ io_dict["inputs"]["weights"] = [
+ 0 for i in range(num_w_reps * n_weight_inps)
+ ]
txns_in["weights"] = []
except AttributeError:
pass
--
GitLab