diff --git a/src/finn/custom_op/fpgadataflow/fmpadding_batch.py b/src/finn/custom_op/fpgadataflow/fmpadding_batch.py index c25975d0c3974a68a8e6ce0502d294406f76cd72..f9a9dc4340b18578550a9c453d90de86234d1cad 100644 --- a/src/finn/custom_op/fpgadataflow/fmpadding_batch.py +++ b/src/finn/custom_op/fpgadataflow/fmpadding_batch.py @@ -47,7 +47,7 @@ class FMPadding_Batch(HLSCustomOp): channels = self.get_nodeattr("NumChannels") simd = self.get_nodeattr("SIMD") batch_size = self.get_nodeattr("numInputVectors") - exp_cycles = channels / simd * batch_size * odim * odim + exp_cycles = (channels / simd) * batch_size * odim * odim return exp_cycles def get_normal_input_shape(self):