From beda40489a25be638bc73dc33428b950fe52ddb9 Mon Sep 17 00:00:00 2001
From: moneta <lorenzo.moneta@cern.ch>
Date: Tue, 14 Aug 2018 18:10:05 +0200
Subject: [PATCH] Fix memory leak in UpdateWeights in Cuda
---
tmva/tmva/src/DNN/Architectures/Cuda/Propagation.cu | 3 +++
1 file changed, 3 insertions(+)
diff --git a/tmva/tmva/src/DNN/Architectures/Cuda/Propagation.cu b/tmva/tmva/src/DNN/Architectures/Cuda/Propagation.cu
index 1674476ab6b..9624016683b 100644
--- a/tmva/tmva/src/DNN/Architectures/Cuda/Propagation.cu
+++ b/tmva/tmva/src/DNN/Architectures/Cuda/Propagation.cu
@@ -385,6 +385,9 @@ void TCuda<AFloat>::CalculateConvWeightGradients(TCudaMatrix<AFloat> & weightGra
// Launch the kernel using our device pointers.
::TMVA::DNN::Cuda::UpdateWeights<<<gridDims, blockDims>>>(weightGradients.GetDataPointer(), dB, batchSize,
depth, nLocalViewPixels);
+
+ delete [] hB;
+ cudaFree(dB);
}
//____________________________________________________________________________
--
GitLab