fix(gpu): more sizes in bench KS

2026-01-08 22:28:01 -05:00 · 2025-11-07 10:09:46 +01:00
parent 97f8ee3a84
commit 4dacfbc0b5
1 changed files with 2 additions and 1 deletions
--- a/tfhe-benchmark/benches/core_crypto/ks_bench.rs
+++ b/tfhe-benchmark/benches/core_crypto/ks_bench.rs
@@ -431,8 +431,9 @@ mod cuda {
                    }


-                    for elements_per_stream in (2..=32u64)
+                    for elements_per_stream_i in (4..=32u64)
                    {
+                        let elements_per_stream = elements_per_stream_i * 16;
                        let plaintext_list = PlaintextList::new(
                            Scalar::ZERO,
                            PlaintextCount(elements_per_stream as usize),