diff --git a/tfhe/examples/utilities/params_to_file.rs b/tfhe/examples/utilities/params_to_file.rs index 6377597f34..81a5d969e9 100644 --- a/tfhe/examples/utilities/params_to_file.rs +++ b/tfhe/examples/utilities/params_to_file.rs @@ -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, }; @@ -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::>();