feat(bench): fix find_optimal_batch to reduce bench time

Author: SouchonTheo

tfhe-benchmark/benches/integer/bench.rs

@@ -3559,27 +3559,27 @@ criterion_group!(
     mul_parallelized,
     div_rem_parallelized,
     bitand_parallelized,
-    bitnot,
-    left_shift_parallelized,
-    rotate_left_parallelized,
-    max_parallelized,
-    eq_parallelized,
-    gt_parallelized,
-    if_then_else_parallelized,
-    flip_parallelized,
-    neg_parallelized,
-    leading_zeros_parallelized,
-    ilog2_parallelized,
-    scalar_add_parallelized,
-    scalar_mul_parallelized,
-    scalar_div_parallelized,
-    scalar_rem_parallelized,
-    scalar_bitand_parallelized,
-    scalar_left_shift_parallelized,
-    scalar_rotate_left_parallelized,
-    scalar_max_parallelized,
-    scalar_eq_parallelized,
-    scalar_gt_parallelized,
+    // bitnot,
+    // left_shift_parallelized,
+    // rotate_left_parallelized,
+    // max_parallelized,
+    // eq_parallelized,
+    // gt_parallelized,
+    // if_then_else_parallelized,
+    // flip_parallelized,
+    // neg_parallelized,
+    // leading_zeros_parallelized,
+    // ilog2_parallelized,
+    // scalar_add_parallelized,
+    // scalar_mul_parallelized,
+    // scalar_div_parallelized,
+    // scalar_rem_parallelized,
+    // scalar_bitand_parallelized,
+    // scalar_left_shift_parallelized,
+    // scalar_rotate_left_parallelized,
+    // scalar_max_parallelized,
+    // scalar_eq_parallelized,
+    // scalar_gt_parallelized,
 );
 
 criterion_group!(

tfhe-benchmark/src/find_optimal_batch.rs

@@ -7,6 +7,9 @@ struct MeasureConfig {
     pub starting_batch_size: usize,
     pub minimum_time_per_batch: Duration,
     pub duration_threshold_for_long_test: Duration,
+    /// Hard ceiling: once a single batch run takes longer than this, stop the search rather than
+    /// doubling again (which would ~double the time), to avoid runaway 15min+ benches.
+    pub max_time_per_batch: Duration,
 }
 
 impl MeasureConfig {
@@ -17,6 +20,7 @@ impl MeasureConfig {
             starting_batch_size: 1,
             minimum_time_per_batch: Duration::from_secs(3),
             duration_threshold_for_long_test: Duration::from_secs(30),
+            max_time_per_batch: Duration::from_secs(100),
         }
     }
 }
@@ -63,7 +67,7 @@ where
     let target = cores * measure_config.target_ratio;
     let long_target = cores * measure_config.long_target_ratio;
 
-    let mut low;
+    let mut low = measure_config.starting_batch_size;
     let mut high = measure_config.starting_batch_size;
     let mut last_usage = 0.0;
 
@@ -86,12 +90,16 @@ where
             return high;
         }
 
+        if duration >= measure_config.max_time_per_batch {
+            return low;
+        }
+
         let improvement = (usage - last_usage).abs() / last_usage;
         if improvement < 0.05
             && duration >= measure_config.duration_threshold_for_long_test
             && last_usage > long_target
         {
-            return high;
+            return low;
         }
 
         low = high;