chore(ci): add overflowing add and sub to integer benchmarks

zama-ai · Jan 10, 2024 · 77c0532 · 77c0532
1 parent 00ddfde
commit 77c0532
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 0 deletions.
diff --git a/tfhe/benches/integer/bench.rs b/tfhe/benches/integer/bench.rs
@@ -662,7 +662,9 @@ define_server_key_bench_fn!(method_name: smart_right_shift_parallelized, display
 define_server_key_bench_fn!(method_name: smart_left_shift_parallelized, display_name: left_shift);
 
 define_server_key_bench_default_fn!(method_name: add_parallelized, display_name: add);
+define_server_key_bench_default_fn!(method_name: unsigned_overflowing_add_parallelized, display_name: overflowing_add);
 define_server_key_bench_default_fn!(method_name: sub_parallelized, display_name: sub);
+define_server_key_bench_default_fn!(method_name: unsigned_overflowing_sub_parallelized, display_name: overflowing_sub);
 define_server_key_bench_default_fn!(method_name: mul_parallelized, display_name: mul);
 define_server_key_bench_default_fn!(method_name: div_parallelized, display_name: div);
 define_server_key_bench_default_fn!(method_name: rem_parallelized, display_name: modulo);
@@ -806,11 +808,21 @@ define_server_key_bench_scalar_default_fn!(
     display_name: add,
     rng_func: default_scalar
 );
+define_server_key_bench_scalar_default_fn!(
+    method_name: unsigned_overflowing_scalar_add_parallelized,
+    display_name: overflowing_add,
+    rng_func: default_scalar
+);
 define_server_key_bench_scalar_default_fn!(
     method_name: scalar_sub_parallelized,
     display_name: sub,
     rng_func: default_scalar
 );
+define_server_key_bench_scalar_default_fn!(
+    method_name: unsigned_overflowing_scalar_sub_parallelized,
+    display_name: overflowing_sub,
+    rng_func: default_scalar
+);
 define_server_key_bench_scalar_default_fn!(
     method_name: scalar_mul_parallelized,
     display_name: mul,
@@ -1181,7 +1193,9 @@ criterion_group!(
     neg_parallelized,
     abs_parallelized,
     add_parallelized,
+    unsigned_overflowing_add_parallelized,
     sub_parallelized,
+    unsigned_overflowing_sub_parallelized,
     mul_parallelized,
     // div_parallelized,
     // rem_parallelized,
@@ -1246,7 +1260,9 @@ criterion_group!(
 criterion_group!(
     default_scalar_parallelized_ops,
     scalar_add_parallelized,
+    unsigned_overflowing_scalar_add_parallelized,
     scalar_sub_parallelized,
+    unsigned_overflowing_scalar_sub_parallelized,
     scalar_mul_parallelized,
     scalar_div_parallelized,
     scalar_rem_parallelized,

diff --git a/tfhe/benches/integer/signed_bench.rs b/tfhe/benches/integer/signed_bench.rs
@@ -361,10 +361,18 @@ define_server_key_bench_binary_signed_clean_inputs_fn!(
     method_name: add_parallelized,
     display_name: add
 );
+define_server_key_bench_binary_signed_clean_inputs_fn!(
+    method_name: signed_overflowing_add_parallelized,
+    display_name: overflowing_add
+);
 define_server_key_bench_binary_signed_clean_inputs_fn!(
     method_name: sub_parallelized,
     display_name: sub
 );
+define_server_key_bench_binary_signed_clean_inputs_fn!(
+    method_name: signed_overflowing_sub_parallelized,
+    display_name: overflowing_sub
+);
 define_server_key_bench_binary_signed_clean_inputs_fn!(
     method_name: mul_parallelized,
     display_name: mul
@@ -483,7 +491,9 @@ criterion_group!(
     neg_parallelized,
     abs_parallelized,
     add_parallelized,
+    signed_overflowing_add_parallelized,
     sub_parallelized,
+    signed_overflowing_sub_parallelized,
     mul_parallelized,
     // div_parallelized,
     // rem_parallelized,
@@ -511,6 +521,16 @@ criterion_group!(
     signed_if_then_else_parallelized,
 );
 
+define_server_key_bench_binary_signed_clean_inputs_fn!(
+    method_name: unchecked_signed_overflowing_add_parallelized,
+    display_name: overflowing_add
+);
+
+define_server_key_bench_binary_signed_clean_inputs_fn!(
+    method_name: unchecked_signed_overflowing_sub_parallelized,
+    display_name: overflowing_sub
+);
+
 define_server_key_bench_binary_signed_clean_inputs_fn!(
     method_name: unchecked_mul_parallelized,
     display_name: mul
@@ -634,6 +654,8 @@ define_server_key_bench_unary_signed_clean_input_fn!(
 
 criterion_group!(
     unchecked_ops,
+    unchecked_signed_overflowing_add_parallelized,
+    unchecked_signed_overflowing_sub_parallelized,
     unchecked_mul_parallelized,
     unchecked_left_shift_parallelized,
     unchecked_right_shift_parallelized,
@@ -819,11 +841,21 @@ define_server_key_bench_binary_scalar_clean_inputs_fn!(
     display_name: add,
     rng_func: default_scalar
 );
+define_server_key_bench_binary_scalar_clean_inputs_fn!(
+    method_name: signed_overflowing_scalar_add_parallelized,
+    display_name: overflowing_add,
+    rng_func: default_scalar
+);
 define_server_key_bench_binary_scalar_clean_inputs_fn!(
     method_name: scalar_sub_parallelized,
     display_name: sub,
     rng_func: default_scalar
 );
+define_server_key_bench_binary_scalar_clean_inputs_fn!(
+    method_name: signed_overflowing_scalar_sub_parallelized,
+    display_name: overflowing_sub,
+    rng_func: default_scalar
+);
 define_server_key_bench_binary_scalar_clean_inputs_fn!(
     method_name: scalar_mul_parallelized,
     display_name: mul,
@@ -925,7 +957,9 @@ define_server_key_bench_binary_scalar_clean_inputs_fn!(
 criterion_group!(
     default_scalar_parallelized_ops,
     scalar_add_parallelized,
+    signed_overflowing_scalar_add_parallelized,
     scalar_sub_parallelized,
+    signed_overflowing_scalar_sub_parallelized,
     scalar_mul_parallelized,
     signed_scalar_div_parallelized,
     signed_scalar_rem_parallelized, // For scalar rem == div_rem