From 9e18b5ce70453f4a8dfc9facb078ee9cc97f945c Mon Sep 17 00:00:00 2001
From: Agnes Leroy <agnes.leroy@zama.ai>
Date: Thu, 1 Aug 2024 17:28:19 +0200
Subject: [PATCH] refactor(gpu): configure GPU parameters automatically to
 multi-bit

---
 tfhe/src/high_level_api/keys/inner.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/tfhe/src/high_level_api/keys/inner.rs b/tfhe/src/high_level_api/keys/inner.rs
index 0057c5a8b1..960671e42b 100644
--- 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(),