From d3cb23a3f46580723fd9ee4b52bdfd9b94a63ffc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Test=C3=A9?= <david.teste@zama.ai>
Date: Mon, 8 Dec 2025 11:59:37 +0100
Subject: [PATCH] WIP: Add more size to bench zk

---
 tfhe-benchmark/benches/integer/zk_pke.rs | 6 +++---
 tfhe/web_wasm_parallel_tests/worker.js   | 8 ++++----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/tfhe-benchmark/benches/integer/zk_pke.rs b/tfhe-benchmark/benches/integer/zk_pke.rs
index b0b489a0b..9d695f164 100644
--- a/tfhe-benchmark/benches/integer/zk_pke.rs
+++ b/tfhe-benchmark/benches/integer/zk_pke.rs
@@ -33,9 +33,9 @@ impl ProofConfig {
 
 fn default_proof_config() -> Vec<ProofConfig> {
     vec![
-        ProofConfig::new(64usize, &[64usize]),
-        ProofConfig::new(2048, &[4 * 64, 10 * 64, 2048]),
-        ProofConfig::new(4096, &[4096]),
+        // ProofConfig::new(64usize, &[64usize]),
+        ProofConfig::new(2048, &[64usize, 4 * 64, 10 * 64, 2048]),
+        // ProofConfig::new(4096, &[4096]),
     ]
 }
 
diff --git a/tfhe/web_wasm_parallel_tests/worker.js b/tfhe/web_wasm_parallel_tests/worker.js
index e8b40e469..453e616e7 100644
--- a/tfhe/web_wasm_parallel_tests/worker.js
+++ b/tfhe/web_wasm_parallel_tests/worker.js
@@ -694,10 +694,10 @@ async function compactPublicKeyZeroKnowledgeBench() {
 
     // Proof configuration:
     let proof_configs = [
-      { crs_bit_size: 64, bits_to_encrypt: [64] },
-      { crs_bit_size: 640, bits_to_encrypt: [640] },
-      { crs_bit_size: 2048, bits_to_encrypt: [2048, 64 * 4] }, // 64 * 4 is a production use-case
-      { crs_bit_size: 4096, bits_to_encrypt: [4096] },
+      // { crs_bit_size: 64, bits_to_encrypt: [64] },
+      // { crs_bit_size: 640, bits_to_encrypt: [640] },
+      { crs_bit_size: 2048, bits_to_encrypt: [2048, 64 * 10, 64 * 4, 64] }, // 64 * 4 is a production use-case
+      // { crs_bit_size: 4096, bits_to_encrypt: [4096] },
     ];
 
     for (const proof_config of proof_configs) {