.

zama-ai · Apr 18, 2024 · 3b6c448 · 3b6c448
1 parent 07c010d
commit 3b6c448
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backends/tfhe-cuda-backend/cuda/src/pbs/programmable_bootstrap_tbc_multibit.cuh b/backends/tfhe-cuda-backend/cuda/src/pbs/programmable_bootstrap_tbc_multibit.cuh
@@ -363,7 +363,7 @@ __host__ void host_tbc_multi_bit_programmable_bootstrap(
   cudaSetDevice(stream->gpu_index);
 
   if (!lwe_chunk_size)
-    lwe_chunk_size = get_average_lwe_chunk_size(num_samples);
+    lwe_chunk_size = get_lwe_chunk_size(num_samples);
 
   for (uint32_t lwe_offset = 0; lwe_offset < (lwe_dimension / grouping_factor);
        lwe_offset += lwe_chunk_size) {