[rtl vvau]: minor fix to dsp estimation

Xilinx · Mar 28, 2024 · b84de7b · b84de7b
1 parent dfd501e
commit b84de7b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/finn/custom_op/fpgadataflow/rtl/vectorvectoractivation_rtl.py b/src/finn/custom_op/fpgadataflow/rtl/vectorvectoractivation_rtl.py
@@ -144,8 +144,9 @@ def lut_estimation(self):
         return 0
 
     def dsp_estimation(self):
+        P = self.get_nodeattr("PE")
         Q = self.get_nodeattr("SIMD")
-        return int(np.ceil(Q / 3))
+        return int(P * np.ceil(Q / 3))
 
     def instantiate_ip(self, cmd):
         # instantiate the RTL IP