[Pool] add parantheses around exp cycles calc

d2a36378 · Yaman Umuroglu · GitHub · debbe662 · d2a36378
Unverified Commit d2a36378 authored 4 years ago by Yaman Umuroglu Committed by GitHub 4 years ago
--- a/src/finn/custom_op/fpgadataflow/pool_batch.py
+++ b/src/finn/custom_op/fpgadataflow/pool_batch.py
@@ -143,7 +143,7 @@ class Pool_Batch(HLSCustomOp):
        k = self.get_nodeattr("KernelSize")
        odim = self.get_nodeattr("OutImgDim")
        batch_size = self.get_nodeattr("BatchSize")
-        exp_cycles = (ifm_ch * k * k) / pe * odim * odim * batch_size
+        exp_cycles = ((ifm_ch * k * k) / pe) * odim * odim * batch_size
        return int(exp_cycles)

    def get_instream_width(self):