chore(ci): reduce number of keys generated in keycache

The keys/ folder was containing lots of unused keys in tests and thus was eating around 20GB of disk space.
zama-ai · Jun 14, 2024 · f930775 · f930775
1 parent 3af990b
commit f930775
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/tfhe/examples/utilities/generates_test_keys.rs b/tfhe/examples/utilities/generates_test_keys.rs
@@ -14,8 +14,11 @@ use tfhe::shortint::parameters::key_switching::{
     ShortintKeySwitchingParameters, PARAM_KEYSWITCH_1_1_KS_PBS_TO_2_2_KS_PBS,
 };
 use tfhe::shortint::parameters::{
-    ClassicPBSParameters, WopbsParameters, ALL_MULTI_BIT_PARAMETER_VEC, ALL_PARAMETER_VEC,
-    PARAM_MESSAGE_1_CARRY_1_KS_PBS, PARAM_MESSAGE_2_CARRY_2_KS_PBS, PARAM_MESSAGE_3_CARRY_3_KS_PBS,
+    ClassicPBSParameters, WopbsParameters, ALL_MULTI_BIT_PARAMETER_VEC,
+    PARAM_MESSAGE_1_CARRY_1_KS_PBS, PARAM_MESSAGE_1_CARRY_2_KS_PBS, PARAM_MESSAGE_1_CARRY_3_KS_PBS,
+    PARAM_MESSAGE_1_CARRY_4_KS_PBS, PARAM_MESSAGE_1_CARRY_5_KS_PBS, PARAM_MESSAGE_1_CARRY_6_KS_PBS,
+    PARAM_MESSAGE_2_CARRY_1_KS_PBS, PARAM_MESSAGE_2_CARRY_2_KS_PBS, PARAM_MESSAGE_2_CARRY_3_KS_PBS,
+    PARAM_MESSAGE_3_CARRY_1_KS_PBS, PARAM_MESSAGE_3_CARRY_2_KS_PBS, PARAM_MESSAGE_3_CARRY_3_KS_PBS,
     PARAM_MESSAGE_4_CARRY_4_KS_PBS, WOPBS_PARAM_MESSAGE_1_CARRY_1_KS_PBS,
     WOPBS_PARAM_MESSAGE_2_CARRY_2_KS_PBS, WOPBS_PARAM_MESSAGE_3_CARRY_3_KS_PBS,
     WOPBS_PARAM_MESSAGE_4_CARRY_4_KS_PBS,
@@ -96,7 +99,22 @@ fn client_server_keys() {
             [DEFAULT_PARAMETERS, DEFAULT_PARAMETERS_KS_PBS];
         generate_boolean_keys(&BOOLEAN_PARAMS);
     } else {
-        generate_pbs_keys(&ALL_PARAMETER_VEC);
+        const PBS_KEYS: [ClassicPBSParameters; 13] = [
+            PARAM_MESSAGE_1_CARRY_1_KS_PBS,
+            PARAM_MESSAGE_1_CARRY_2_KS_PBS,
+            PARAM_MESSAGE_1_CARRY_3_KS_PBS,
+            PARAM_MESSAGE_1_CARRY_4_KS_PBS,
+            PARAM_MESSAGE_1_CARRY_5_KS_PBS,
+            PARAM_MESSAGE_1_CARRY_6_KS_PBS,
+            PARAM_MESSAGE_2_CARRY_1_KS_PBS,
+            PARAM_MESSAGE_2_CARRY_2_KS_PBS,
+            PARAM_MESSAGE_2_CARRY_3_KS_PBS,
+            PARAM_MESSAGE_3_CARRY_1_KS_PBS,
+            PARAM_MESSAGE_3_CARRY_2_KS_PBS,
+            PARAM_MESSAGE_3_CARRY_3_KS_PBS,
+            PARAM_MESSAGE_4_CARRY_4_KS_PBS,
+        ];
+        generate_pbs_keys(&PBS_KEYS);
 
         const WOPBS_PARAMS: [(ClassicPBSParameters, WopbsParameters); 4] = [
             (