github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-07 22:04:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(bench): remove trivial encryptions

Browse Source 
It makes benches not accurate
This commit is contained in:
Thomas Montaigu
2025-10-16 09:59:44 +02:00
committed by tmontaigu
parent 5d5e9d47e9
commit 0dd0ead4e2
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tfhe-benchmark/benches/integer/bench.rs
View File

@@ -619,7 +619,7 @@ fn if_then_else_parallelized(c: &mut Criterion) {
let clear_1 = gen_random_u256(&mut rng);
let false_ct = cks.encrypt_radix(clear_1, num_block);
let condition = sks.create_trivial_boolean_block(rng.gen_bool(0.5));
let condition = cks.encrypt_bool(rng.gen_bool(0.5));
reset_pbs_count();
sks.if_then_else_parallelized(&condition, &true_ct, &false_ct);
@@ -634,7 +634,7 @@ fn if_then_else_parallelized(c: &mut Criterion) {
bench_group.bench_function(&bench_id, |b| {
let setup_encrypted_values = || {
let cts_cond = (0..elements)
.map(|_| sks.create_trivial_boolean_block(rng.gen_bool(0.5)))
.map(|_| cks.encrypt_bool(rng.gen_bool(0.5)))
.collect::<Vec<_>>();
let cts_then = (0..elements)
@@ -727,7 +727,7 @@ fn flip_parallelized(c: &mut Criterion) {
let clear_1 = gen_random_u256(&mut rng);
let false_ct = cks.encrypt_radix(clear_1, num_block);
let condition = sks.create_trivial_boolean_block(rng.gen_bool(0.5));
let condition = cks.encrypt_bool(rng.gen_bool(0.5));
reset_pbs_count();
sks.flip_parallelized(&condition, &true_ct, &false_ct);
@@ -742,7 +742,7 @@ fn flip_parallelized(c: &mut Criterion) {
bench_group.bench_function(&bench_id, |b| {
let setup_encrypted_values = || {
let cts_cond = (0..elements)
.map(|_| sks.create_trivial_boolean_block(rng.gen_bool(0.5)))
.map(|_| cks.encrypt_bool(rng.gen_bool(0.5)))
.collect::<Vec<_>>();
let cts_then = (0..elements)
@@ -1961,7 +1961,7 @@ mod cuda {
let ct_then = cks.encrypt_radix(clear_0, num_block);
let clear_1 = gen_random_u256(&mut rng);
let ct_else = cks.encrypt_radix(clear_1, num_block);
let ct_cond = cpu_sks.create_trivial_boolean_block(rng.gen_bool(0.5));
let ct_cond = cks.encrypt_bool(rng.gen_bool(0.5));
reset_pbs_count();
// Use CPU operation as pbs_count do not count PBS on GPU backend.

8
tfhe-benchmark/benches/integer/signed_bench.rs
View File

@@ -325,7 +325,7 @@ fn signed_if_then_else_parallelized(c: &mut Criterion) {
let bench_data = LazyCell::new(|| {
let (cks, sks) = KEY_CACHE.get_from_params(param, IntegerKeyKind::Radix);
let condition = sks.create_trivial_boolean_block(rng.gen_bool(0.5));
let condition = cks.encrypt_bool(rng.gen_bool(0.5));
let true_ct = cks.encrypt_signed_radix(gen_random_i256(&mut rng), num_block);
let false_ct = cks.encrypt_signed_radix(gen_random_i256(&mut rng), num_block);
@@ -344,7 +344,7 @@ fn signed_if_then_else_parallelized(c: &mut Criterion) {
let (cks, sks) = KEY_CACHE.get_from_params(param, IntegerKeyKind::Radix);
// Execute the operation once to know its cost.
let cond = sks.create_trivial_boolean_block(rng.gen_bool(0.5));
let cond = cks.encrypt_bool(rng.gen_bool(0.5));
let ct_then = cks.encrypt_signed_radix(gen_random_i256(&mut rng), num_block);
let ct_else = cks.encrypt_signed_radix(gen_random_i256(&mut rng), num_block);
@@ -358,7 +358,7 @@ fn signed_if_then_else_parallelized(c: &mut Criterion) {
bench_group.bench_function(&bench_id, |b| {
let setup_encrypted_values = || {
let cts_cond = (0..elements)
.map(|_| sks.create_trivial_boolean_block(rng.gen_bool(0.5)))
.map(|_| cks.encrypt_bool(rng.gen_bool(0.5)))
.collect::<Vec<_>>();
let cts_then = (0..elements)
@@ -2113,7 +2113,7 @@ mod cuda {
let gpu_sks_vec = cuda_local_keys(&cks);
// Execute the operation once to know its cost.
let cond = cpu_sks.create_trivial_boolean_block(rng.gen_bool(0.5));
let cond = cks.encrypt_bool(rng.gen_bool(0.5));
let ct_then = cks.encrypt_signed_radix(gen_random_i256(&mut rng), num_block);
let ct_else = cks.encrypt_signed_radix(gen_random_i256(&mut rng), num_block);