[PixelPadding] Add batchsize for expected cycles calc

Xilinx · Jan 3, 2024 · c741fae · c741fae
1 parent c8b80df
commit c741fae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/finn/custom_op/fpgadataflow/fmpadding_pixel.py b/src/finn/custom_op/fpgadataflow/fmpadding_pixel.py
@@ -70,7 +70,8 @@ def get_exp_cycles(self):
         odim_h, odim_w = self.get_padded_odim()
         channels = self.get_nodeattr("NumChannels")
         simd = self.get_nodeattr("SIMD")
-        exp_cycles = (channels / simd) * odim_h * odim_w
+        batch_size = self.get_nodeattr("numInputVectors")
+        exp_cycles = (channels / simd) * batch_size * odim_h * odim_w
         return int(exp_cycles)
 
     def get_normal_input_shape(self, ind=0):