refactor(gpu): configure GPU parameters automatically to multi-bit

zama-ai · Aug 1, 2024 · 9e18b5c · 9e18b5c
1 parent 5547d92
commit 9e18b5c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tfhe/src/high_level_api/keys/inner.rs b/tfhe/src/high_level_api/keys/inner.rs
@@ -42,6 +42,7 @@ impl IntegerConfig {
         }
     }
 
+    #[cfg(not(feature = "gpu"))]
     pub(in crate::high_level_api) fn default_big() -> Self {
         Self {
             block_parameters: crate::shortint::parameters::PARAM_MESSAGE_2_CARRY_2_KS_PBS.into(),
@@ -50,6 +51,17 @@ impl IntegerConfig {
         }
     }
 
+    #[cfg(feature = "gpu")]
+    pub(in crate::high_level_api) fn default_big() -> Self {
+        Self {
+            block_parameters:
+                crate::shortint::parameters::PARAM_GPU_MULTI_BIT_MESSAGE_2_CARRY_2_GROUP_3_KS_PBS
+                    .into(),
+            dedicated_compact_public_key_parameters: None,
+            compression_parameters: None,
+        }
+    }
+
     pub(in crate::high_level_api) fn default_small() -> Self {
         Self {
             block_parameters: crate::shortint::parameters::PARAM_MESSAGE_2_CARRY_2_PBS_KS.into(),