From 4dacfbc0b589e89481d35b75ad93294d00db67b1 Mon Sep 17 00:00:00 2001 From: Andrei Stoian Date: Fri, 7 Nov 2025 10:09:46 +0100 Subject: [PATCH] fix(gpu): more sizes in bench KS --- tfhe-benchmark/benches/core_crypto/ks_bench.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tfhe-benchmark/benches/core_crypto/ks_bench.rs b/tfhe-benchmark/benches/core_crypto/ks_bench.rs index 66d5c923a..20abb1663 100644 --- 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),