From aa16aaf3a095d27b3e1722b80d31db1fdada5199 Mon Sep 17 00:00:00 2001 From: Agnes Leroy Date: Mon, 16 Dec 2024 10:29:10 +0100 Subject: [PATCH] DO NOT MERGE: remove device synchronization in drop for CudaVec to check the effect on multi-gpu throughput benchmarks --- tfhe/src/core_crypto/gpu/vec.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tfhe/src/core_crypto/gpu/vec.rs b/tfhe/src/core_crypto/gpu/vec.rs index 8063457f53..7863559974 100644 --- a/tfhe/src/core_crypto/gpu/vec.rs +++ b/tfhe/src/core_crypto/gpu/vec.rs @@ -449,7 +449,7 @@ impl Drop for CudaVec { .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) }; } }