diff --git a/backends/tfhe-cuda-backend/cuda/include/integer/oprf.h b/backends/tfhe-cuda-backend/cuda/include/integer/oprf.h index 9b283f23f..ecd2eab9f 100644 --- 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 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); }