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];