fix(gpu): add missing sync before free in oprf

2026-01-08 06:13:58 -05:00 · 2025-12-15 16:45:05 +01:00
parent 006d6cc300
commit cfa53682ae
1 changed files with 1 additions and 0 deletions
--- a/backends/tfhe-cuda-backend/cuda/include/integer/oprf.h
+++ b/backends/tfhe-cuda-backend/cuda/include/integer/oprf.h
@@ -129,6 +129,7 @@ template <typename Torus> struct int_grouped_oprf_memory {
    auto active_streams = streams.active_gpu_subset(num_blocks_to_process);
    luts->broadcast_lut(active_streams);

+    cuda_synchronize_stream(streams.stream(0), streams.gpu_index(0));
    free(h_corrections);
  }