github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-09 14:47:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(gpu): change a type to avoid possible issues when compression is executed for a large batch of LWEs

Browse Source
This commit is contained in:
Pedro Alves
2025-11-24 10:42:40 -03:00
parent 0b98ef98fc
commit 758a065fc4
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backends/tfhe-cuda-backend/cuda/src/integer/compression/compression.cuh
View File

@@ -136,9 +136,9 @@ host_integer_compress(CudaStreams streams,
} }
// Modulus switch // Modulus switch
int size = num_glwes * compression_params.glwe_dimension * uint32_t size = num_glwes * compression_params.glwe_dimension *
compression_params.polynomial_size + compression_params.polynomial_size +
glwe_array_out->total_lwe_bodies_count; glwe_array_out->total_lwe_bodies_count;
host_modulus_switch_inplace<Torus>(streams.stream(0), streams.gpu_index(0), host_modulus_switch_inplace<Torus>(streams.stream(0), streams.gpu_index(0),
tmp_glwe_array_out, size, tmp_glwe_array_out, size,