diff --git a/backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cuh b/backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cuh index 826580d565..35842fcc11 100644 --- a/backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cuh +++ b/backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cuh @@ -136,9 +136,9 @@ host_integer_compress(CudaStreams streams, } // Modulus switch - int size = num_glwes * compression_params.glwe_dimension * - compression_params.polynomial_size + - glwe_array_out->total_lwe_bodies_count; + uint32_t size = num_glwes * compression_params.glwe_dimension * + compression_params.polynomial_size + + glwe_array_out->total_lwe_bodies_count; host_modulus_switch_inplace(streams.stream(0), streams.gpu_index(0), tmp_glwe_array_out, size,