From 8c32a78fca885dcb65724940b8f1e999665827ab Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Mon, 27 Jul 2020 14:34:56 +0100
Subject: [PATCH] [CustomOp] Change calculation of exp cycles for GlobalAccPool
---
src/finn/custom_op/fpgadataflow/globalaccpool_batch.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/globalaccpool_batch.py b/src/finn/custom_op/fpgadataflow/globalaccpool_batch.py
index 477fcf507..7bfd4eb59 100644
--- a/src/finn/custom_op/fpgadataflow/globalaccpool_batch.py
+++ b/src/finn/custom_op/fpgadataflow/globalaccpool_batch.py
@@ -183,8 +183,8 @@ class GlobalAccPool_Batch(HLSCustomOp):
return np.prod(self.get_folded_output_shape()[1:-1])
def get_exp_cycles(self):
- # Channels/PE * batch size (ofmdim = 1)
- return np.prod(self.get_folded_output_shape()[:-1])
+ # Channels/PE * batch size * idim * idim
+ return np.prod(self.get_folded_input_shape()[:-1])
def execute_node(self, context, graph):
mode = self.get_nodeattr("exec_mode")
--
GitLab