From 18d17bfde7af81787ea046f55b757a04a16a0279 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Tue, 23 Nov 2021 11:00:41 +0100
Subject: [PATCH] [Pool] adjust kernel param as total k
---
src/finn/custom_op/fpgadataflow/pool_batch.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/finn/custom_op/fpgadataflow/pool_batch.py b/src/finn/custom_op/fpgadataflow/pool_batch.py
index ba8a446f2..92679e306 100644
--- a/src/finn/custom_op/fpgadataflow/pool_batch.py
+++ b/src/finn/custom_op/fpgadataflow/pool_batch.py
@@ -211,7 +211,7 @@ class Pool_Batch(HLSCustomOp):
self.code_gen_dict["$DEFINES$"] += ["#define PE {}".format(pe)]
k = self.get_nodeattr("KernelSize")
- self.code_gen_dict["$DEFINES$"] += ["#define KernelSize {}".format(k)]
+ self.code_gen_dict["$DEFINES$"] += ["#define KernelSize {}".format(k * k)]
odim = self.get_nodeattr("OutImgDim")
self.code_gen_dict["$DEFINES$"] += ["#define OFMDim {}".format(odim)]
--
GitLab