[Folding] Change condition for meeting target cycles from '<' to '<='

eki-project · May 3, 2024 · d320ac0 · d320ac0
1 parent 766e9fb
commit d320ac0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/finn/transformation/fpgadataflow/set_folding.py b/src/finn/transformation/fpgadataflow/set_folding.py
@@ -92,7 +92,7 @@ def optimize_attribute_val(self, node_inst, max_val, attr_name):
         for val in divisors(max_val):
             node_inst.set_nodeattr(attr_name, val)
             cyc = node_inst.get_exp_cycles()
-            if cyc < self.target_cycles_per_frame:
+            if cyc <= self.target_cycles_per_frame:
                 # finish if target met
                 break
 
@@ -138,7 +138,7 @@ def apply(self, model):
                     prev_simd_val = node_inst.get_nodeattr("SIMD")
                     node_inst.set_nodeattr("SIMD", simd_val)
                     cyc = node_inst.get_exp_cycles()
-                    if cyc < self.target_cycles_per_frame:
+                    if cyc <= self.target_cycles_per_frame:
                         # finish if target met
                         break
                     if (