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

feat(hpu-noise) cleanup a few warnings and reduce amount of data printed

Browse Source
This commit is contained in:
pgardratzama
2024-11-29 10:24:29 +01:00
committed by tmontaigu
parent 3c5e141b9d
commit ebbab2991e
1 changed files with 3 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tfhe/src/core_crypto/algorithms/test/noise_distribution/lwe_encryption_noise.rs
View File

@@ -315,6 +315,7 @@ fn hpu_noise_distribution(params: HpuTestParams) {
ksk_modulus,
&mut rsc.encryption_random_generator,
);
println!(
"n {:?} k {:?} N {:?} k*N {:?}",
lwe_sk.lwe_dimension(),
@@ -324,7 +325,7 @@ fn hpu_noise_distribution(params: HpuTestParams) {
);
// it includes variance of mod switch from KS modulus to 2N
let (exp_add_ks_variance, exp_modswitch_variance) =
let (exp_add_ks_variance, _exp_modswitch_variance) =
variance_formula::lwe_keyswitch::keyswitch_additive_variance_128_bits_security_gaussian(
glwe_dimension,
polynomial_size,
@@ -358,7 +359,7 @@ fn hpu_noise_distribution(params: HpuTestParams) {
&mut rsc.encryption_random_generator,
);
let mut exp_pbs_variance =
let exp_pbs_variance =
variance_formula::lwe_programmable_bootstrap::pbs_variance_128_bits_security_gaussian(
lwe_dimension,
glwe_dimension,
@@ -440,10 +441,6 @@ fn hpu_noise_distribution(params: HpuTestParams) {
assert_eq!(msg, decoded);
let torus_diff = torus_modular_diff(plaintext.0, decrypted.0, ciphertext_modulus);
println!(
"after decryption: plaintext {:?} decrypted {:?} torus_diff {:?}",
plaintext.0, decrypted.0, torus_diff
);
noise_samples[0].push(torus_diff);
for j in 0..NB_PBS {
@@ -469,10 +466,6 @@ fn hpu_noise_distribution(params: HpuTestParams) {
let decode_prodnorm2 = round_decode(decrypted_prodnorm2.0, delta) % msg_modulus;
let torus_diff = torus_modular_diff(0, decrypted_prodnorm2.0, ciphertext_modulus);
println!(
"after by norm2: plaintext {:?} bynorm2 {:?} torus_diff {:?}",
0, decrypted_prodnorm2.0, torus_diff
);
assert_eq!(0, decode_prodnorm2);
noise_samples[1].push(torus_diff);
// b = b + (Delta * msg) to have a noisy encryption of msg
@@ -506,10 +499,6 @@ fn hpu_noise_distribution(params: HpuTestParams) {
if cm_f == ksm_f {
torus_diff =
torus_modular_diff(plaintext.0, decrypted_after_ks.0, ciphertext_modulus);
println!(
"after ks {} / {} vs {}",
torus_diff, plaintext.0, decrypted_after_ks.0
);
} else {
let decrypted_after_ks_modswitched =
decrypted_after_ks.0 * ((cm_f / ksm_f) as u64);
@@ -518,13 +507,6 @@ fn hpu_noise_distribution(params: HpuTestParams) {
decrypted_after_ks_modswitched,
ciphertext_modulus,
);
println!(
"after ks {} / {} vs {}*CT_MOD/KS_MOD={}",
torus_diff,
plaintext.0,
decrypted_after_ks.0,
decrypted_after_ks_modswitched
);
}
noise_samples[2].push(torus_diff);