DO NOT MERGE: remove device synchronization in drop for CudaVec to ch…

…eck the effect on multi-gpu throughput benchmarks
zama-ai · Dec 16, 2024 · aa16aaf · aa16aaf
1 parent 3a99ee9
commit aa16aaf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tfhe/src/core_crypto/gpu/vec.rs b/tfhe/src/core_crypto/gpu/vec.rs
@@ -449,7 +449,7 @@ impl<T: Numeric> Drop for CudaVec<T> {
             .zip(self.gpu_indexes.iter().copied())
         {
             // Synchronizes the device to be sure no stream is still using this pointer
-            synchronize_device(gpu_index.0);
+            //synchronize_device(gpu_index.0);
             unsafe { cuda_drop(ptr, gpu_index.0) };
         }
     }