Skip to content

Commit

Permalink
chore(ci): write gpu parameters to file
Browse files Browse the repository at this point in the history
This is done so that lattice estimator can check security of these
ones.
  • Loading branch information
soonum committed Sep 26, 2024
1 parent 8c51e22 commit ea3ec8c
Showing 1 changed file with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion tfhe/examples/utilities/params_to_file.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ use tfhe::keycache::NamedParam;
use tfhe::shortint::parameters::classic::compact_pk::ALL_PARAMETER_VEC_COMPACT_PK;
use tfhe::shortint::parameters::classic::gaussian::ALL_PARAMETER_VEC_GAUSSIAN;
use tfhe::shortint::parameters::multi_bit::ALL_MULTI_BIT_PARAMETER_VEC;
use tfhe::shortint::parameters::p_fail_2_minus_64::ks_pbs_gpu::*;
use tfhe::shortint::parameters::{
ShortintParameterSet, PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M64,
};
Expand Down Expand Up @@ -198,7 +199,19 @@ fn main() {
&classic_pbs,
);

let multi_bit_pbs = ALL_MULTI_BIT_PARAMETER_VEC
let all_multi_pbs = [
ALL_MULTI_BIT_PARAMETER_VEC.to_vec(),
vec![
PARAM_GPU_MULTI_BIT_GROUP_2_MESSAGE_1_CARRY_1_KS_PBS_GAUSSIAN_2M64,
PARAM_GPU_MULTI_BIT_GROUP_2_MESSAGE_2_CARRY_2_KS_PBS_GAUSSIAN_2M64,
PARAM_GPU_MULTI_BIT_GROUP_2_MESSAGE_3_CARRY_3_KS_PBS_GAUSSIAN_2M64,
PARAM_GPU_MULTI_BIT_GROUP_3_MESSAGE_1_CARRY_1_KS_PBS_GAUSSIAN_2M64,
PARAM_GPU_MULTI_BIT_GROUP_3_MESSAGE_2_CARRY_2_KS_PBS_GAUSSIAN_2M64,
PARAM_GPU_MULTI_BIT_GROUP_3_MESSAGE_3_CARRY_3_KS_PBS_GAUSSIAN_2M64,
],
]
.concat();
let multi_bit_pbs = all_multi_pbs
.iter()
.map(|p| ShortintParameterSet::from(*p))
.collect::<Vec<_>>();
Expand Down

0 comments on commit ea3ec8c

Please sign in to comment.