From 8c894ed550da7600758ba2fafad10867f0210242 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Fri, 22 Jul 2022 19:43:25 +0200
Subject: [PATCH] [VVAU] bugfix in weight reps for node-by-node rtlsim
---
src/finn/custom_op/fpgadataflow/vectorvectoractivation.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py b/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py
index 24cb3101f..77fed5e3a 100644
--- a/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py
+++ b/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py
@@ -670,7 +670,9 @@ class VectorVectorActivation(HLSCustomOp):
wei = npy_to_rtlsim_input(
"{}/weights.npy".format(code_gen_dir), export_wdt, wnbits
)
- num_w_reps = 1
+ dim_h, dim_w = self.get_nodeattr("Dim")
+ num_w_reps = dim_h * dim_w
+
io_dict = {
"inputs": {"in0": inp, "weights": wei * num_w_reps},
"outputs": {"out": []},
--
GitLab