fix(gpu): fix memory leak

zama-ai · Dec 17, 2024 · 072005d · 072005d
1 parent 241b737
commit 072005d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backends/tfhe-cuda-backend/cuda/src/integer/comparison.cuh b/backends/tfhe-cuda-backend/cuda/src/integer/comparison.cuh
@@ -137,6 +137,8 @@ __host__ void are_all_comparisons_block_true(
                                  h_lut_indexes, num_chunks * sizeof(Torus),
                                  streams[0], gpu_indexes[0]);
         is_max_value_lut->broadcast_lut(streams, gpu_indexes, 0);
+        cuda_synchronize_stream(streams[0], gpu_indexes[0]);
+        free(h_lut_indexes);
       }
       lut = is_max_value_lut;
     }