github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-10 07:08:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(gpu): add missing delete

Browse Source
This commit is contained in:
Agnes Leroy
2024-07-16 09:52:56 +02:00
committed by Agnès Leroy
parent ebf1fd9e84
commit abf9c3efb7
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backends/tfhe-cuda-backend/cuda/include/integer.h
View File

@@ -1035,6 +1035,7 @@ template <typename Torus> struct int_fullprop_buffer {
uint32_t gpu_count) {
lut->release(streams, gpu_indexes, 1);
delete lut;
cuda_drop_async(tmp_small_lwe_vector, streams[0], gpu_indexes[0]);
cuda_drop_async(tmp_big_lwe_vector, streams[0], gpu_indexes[0]);
@@ -3258,6 +3259,7 @@ template <typename Torus> struct int_scalar_mul_buffer {
void release(cudaStream_t *streams, uint32_t *gpu_indexes,
uint32_t gpu_count) {
sum_ciphertexts_vec_mem->release(streams, gpu_indexes, gpu_count);
delete sum_ciphertexts_vec_mem;
cuda_drop_async(all_shifted_buffer, streams[0], gpu_indexes[0]);
}
};