From a5fde5832050fbb7d3cfa411a985fd039e6aa552 Mon Sep 17 00:00:00 2001 From: Agnes Leroy Date: Tue, 16 Jul 2024 09:52:56 +0200 Subject: [PATCH] fix(gpu): add missing delete --- backends/tfhe-cuda-backend/cuda/include/integer.h | 2 ++ 1 file changed, 2 insertions(+) diff --git a/backends/tfhe-cuda-backend/cuda/include/integer.h b/backends/tfhe-cuda-backend/cuda/include/integer.h index f7ee72f9db..af57505331 100644 --- a/backends/tfhe-cuda-backend/cuda/include/integer.h +++ b/backends/tfhe-cuda-backend/cuda/include/integer.h @@ -1035,6 +1035,7 @@ template 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 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]); } };