Merge pull request #16 from zama-ai/petar/update-tfhe-rs

Update tfhe-rs to 0.3.1
zama-ai · Aug 23, 2023 · 576fee8 · 576fee8
2 parents 7fd4b41 + 5c928de
commit 576fee8
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -12,4 +12,4 @@ publish = false
 bincode = "1.3.3"
 clap = { version = "3.1.18", features = ["derive"] }
 serde = "1.0"
-tfhe = { git = "https://github.com/zama-ai/tfhe-rs.git", rev = "0.3.0-beta.0", features = ["boolean", "experimental-force_fft_algo_dif4", "shortint", "integer"] }
+tfhe = { git = "https://github.com/zama-ai/tfhe-rs.git", rev = "0.3.1", features = ["boolean", "experimental-force_fft_algo_dif4", "shortint", "integer"] }
diff --git a/src/gen_keys.rs b/src/gen_keys.rs
@@ -4,13 +4,14 @@
 // All rights reserved.
 
 use tfhe::{
-    generate_keys, shortint::parameters::PARAM_SMALL_MESSAGE_2_CARRY_2_COMPACT_PK, ClientKey,
-    CompactPublicKey, ConfigBuilder, ServerKey,
+    generate_keys,
+    shortint::parameters::parameters_compact_pk::PARAM_MESSAGE_2_CARRY_2_COMPACT_PK_PBS_KS,
+    ClientKey, CompactPublicKey, ConfigBuilder, ServerKey,
 };
 
 pub fn gen_keys() -> (ClientKey, ServerKey, CompactPublicKey) {
     let config = ConfigBuilder::all_disabled()
-        .enable_custom_integers(PARAM_SMALL_MESSAGE_2_CARRY_2_COMPACT_PK, None)
+        .enable_custom_integers(PARAM_MESSAGE_2_CARRY_2_COMPACT_PK_PBS_KS, None)
         .build();
     let (cks, sks) = generate_keys(config);
     let pks = CompactPublicKey::new(&cks);