chore(zk): use pke v2 for zk

zama-ai · Sep 9, 2024 · 0f93d31 · 0f93d31
1 parent 6230ffa
commit 0f93d31
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tfhe/src/core_crypto/algorithms/lwe_encryption.rs b/tfhe/src/core_crypto/algorithms/lwe_encryption.rs
@@ -16,7 +16,7 @@ use crate::core_crypto::commons::traits::*;
 use crate::core_crypto::entities::*;
 use rayon::prelude::*;
 #[cfg(feature = "zk-pok")]
-use tfhe_zk_pok::proofs::pke::{commit, prove};
+use tfhe_zk_pok::proofs::pke_v2::{commit, prove};
 
 /// Convenience function to share the core logic of the LWE encryption between all functions needing
 /// it.

diff --git a/tfhe/src/core_crypto/algorithms/lwe_zero_knowledge_verification.rs b/tfhe/src/core_crypto/algorithms/lwe_zero_knowledge_verification.rs
@@ -1,7 +1,7 @@
 use crate::core_crypto::entities::{LweCompactCiphertextList, LweCompactPublicKey};
 use crate::core_crypto::prelude::{CastFrom, Container, LweCiphertext, UnsignedInteger};
 use crate::zk::{CompactPkeProof, CompactPkePublicParams, ZkVerificationOutCome};
-use tfhe_zk_pok::proofs::pke::{verify, PublicCommit};
+use tfhe_zk_pok::proofs::pke_v2::{verify, PublicCommit};
 
 /// Verifies with the given proof that a [`LweCompactCiphertextList`]
 /// is valid.

diff --git a/tfhe/src/zk.rs b/tfhe/src/zk.rs
@@ -4,7 +4,7 @@ use rand_core::RngCore;
 use std::cmp::Ordering;
 use std::collections::Bound;
 use std::fmt::Debug;
-use tfhe_zk_pok::proofs::pke::crs_gen;
+use tfhe_zk_pok::proofs::pke_v2::crs_gen;
 
 pub use tfhe_zk_pok::proofs::ComputeLoad as ZkComputeLoad;
 pub use tfhe_zk_pok::{CanonicalDeserialize, CanonicalSerialize, Compress, Validate};