chore(bench): bench 64 bits for ZKs

zama-ai · Sep 26, 2024 · 5d0742c · 5d0742c
1 parent 4bb115e
commit 5d0742c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tfhe/benches/integer/zk_pke.rs b/tfhe/benches/integer/zk_pke.rs
@@ -53,7 +53,7 @@ fn pke_zk_proof(c: &mut Criterion) {
         let mut rng = rand::thread_rng();
         metadata.fill_with(|| rng.gen());
 
-        for bits in [640usize, 1280, 4096] {
+        for bits in [64usize, 640, 1280, 4096] {
             assert_eq!(bits % 64, 0);
             // Packing, so we take the message and carry modulus to compute our block count
             let num_block = 64usize.div_ceil(
@@ -138,7 +138,7 @@ fn pke_zk_verify(c: &mut Criterion, results_file: &Path) {
         let mut rng = rand::thread_rng();
         metadata.fill_with(|| rng.gen());
 
-        for bits in [640usize, 1280, 4096] {
+        for bits in [64usize, 640, 1280, 4096] {
             assert_eq!(bits % 64, 0);
             // Packing, so we take the message and carry modulus to compute our block count
             let num_block = 64usize.div_ceil(

diff --git a/tfhe/web_wasm_parallel_tests/worker.js b/tfhe/web_wasm_parallel_tests/worker.js
@@ -669,7 +669,7 @@ async function compactPublicKeyZeroKnowledgeBench() {
       [ZkComputeLoad.Verify]: "compute_load_verify",
     };
 
-    let bits_to_encrypt = [640, 1280, 4096];
+    let bits_to_encrypt = [64, 640, 1280, 4096];
 
     let encrypt_counts = bits_to_encrypt.map((v) => v / 64);