github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-09 14:47:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(shortint): add support for centered modulus switch in parameters

Browse Source
This commit is contained in:
Mayeul@Zama
2025-06-19 14:49:07 +02:00
committed by Nicolas Sarlin
parent 6805778cb8
commit e1620d4087
73 changed files with 2499 additions and 1701 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
tests/backward_compatibility/shortint.rs
View File

@@ -2,13 +2,14 @@ use std::path::Path;
use tfhe::core_crypto::prelude::{
LweCiphertextCount, NoiseEstimationMeasureBound, RSigmaFactor, TUniform, Variance,
};
use tfhe::shortint::parameters::ModulusSwitchNoiseReductionParams;
use tfhe::shortint::parameters::{ModulusSwitchNoiseReductionParams, ModulusSwitchType};
use tfhe_backward_compat_data::load::{
load_versioned_auxiliary, DataFormat, TestFailure, TestResult, TestSuccess,
};
use tfhe_backward_compat_data::{
ShortintCiphertextTest, ShortintClientKeyTest, TestDistribution, TestMetadata,
TestModulusSwitchNoiseReductionParams, TestParameterSet, TestType, Testcase,
TestModulusSwitchNoiseReductionParams, TestModulusSwitchType, TestParameterSet, TestType,
Testcase,
};
use tfhe::shortint::parameters::{
@@ -45,21 +46,25 @@ pub fn load_params(test_params: &TestParameterSet) -> ClassicPBSParameters {
modulus_switch_noise_reduction_params,
} = test_params;
let modulus_switch_noise_reduction_params = modulus_switch_noise_reduction_params.as_ref().map(
|TestModulusSwitchNoiseReductionParams {
modulus_switch_zeros_count,
ms_bound,
ms_r_sigma_factor,
ms_input_variance,
}| {
ModulusSwitchNoiseReductionParams {
modulus_switch_zeros_count: LweCiphertextCount(*modulus_switch_zeros_count),
ms_bound: NoiseEstimationMeasureBound(*ms_bound),
ms_r_sigma_factor: RSigmaFactor(*ms_r_sigma_factor),
ms_input_variance: Variance(*ms_input_variance),
}
},
);
let modulus_switch_noise_reduction_params = match modulus_switch_noise_reduction_params {
TestModulusSwitchType::Standard => ModulusSwitchType::Standard,
TestModulusSwitchType::DriftTechniqueNoiseReduction(
TestModulusSwitchNoiseReductionParams {
modulus_switch_zeros_count,
ms_bound,
ms_r_sigma_factor,
ms_input_variance,
},
) => ModulusSwitchType::DriftTechniqueNoiseReduction(ModulusSwitchNoiseReductionParams {
modulus_switch_zeros_count: LweCiphertextCount(*modulus_switch_zeros_count),
ms_bound: NoiseEstimationMeasureBound(*ms_bound),
ms_r_sigma_factor: RSigmaFactor(*ms_r_sigma_factor),
ms_input_variance: Variance(*ms_input_variance),
}),
TestModulusSwitchType::CenteredMeanNoiseReduction => {
ModulusSwitchType::CenteredMeanNoiseReduction
}
};
ClassicPBSParameters {
lwe_dimension: LweDimension(*lwe_dimension),