diff --git a/src/finn/transformation/fpgadataflow/set_folding.py b/src/finn/transformation/fpgadataflow/set_folding.py index eaee499e6a..3ce17a27ff 100644 --- 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 (