From 89b36ebca014a4216fc8f0f99e9bc6694cbb9db9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20Test=C3=A9?= Date: Mon, 8 Sep 2025 09:49:20 +0200 Subject: [PATCH] chore(bench): remove 2-bits size for full precision bench on gpu GPU backend cannot accept less than 2 blocks for integer benchmarks. Since 2-bits precision benchmarks are run with *_MESSAGE_2_CARRY_2_* parameters, it will create only one block of ciphertext, thus making the benchmarks unsuitable for GPU backend. --- tfhe-benchmark/src/utilities.rs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tfhe-benchmark/src/utilities.rs b/tfhe-benchmark/src/utilities.rs index 0d43e4c0d..4ce028e80 100644 --- a/tfhe-benchmark/src/utilities.rs +++ b/tfhe-benchmark/src/utilities.rs @@ -348,7 +348,10 @@ pub fn write_to_json< } const FAST_BENCH_BIT_SIZES: [usize; 1] = [64]; +#[cfg(not(feature = "gpu"))] const BENCH_BIT_SIZES: [usize; 8] = [2, 8, 16, 32, 40, 64, 128, 256]; +#[cfg(feature = "gpu")] +const BENCH_BIT_SIZES: [usize; 7] = [8, 16, 32, 40, 64, 128, 256]; const HPU_BENCH_BIT_SIZES: [usize; 5] = [8, 16, 32, 64, 128]; const MULTI_BIT_CPU_SIZES: [usize; 5] = [8, 16, 32, 40, 64];