From 7dfbc544c3a6f986a51d3d85f055e7ee671a3cb3 Mon Sep 17 00:00:00 2001 From: Tomohiro Ubukata Date: Sun, 1 Mar 2020 08:57:58 +0000 Subject: [PATCH] Fix CuDNNLRNLayer Resource leak --- src/caffe/layers/cudnn_lrn_layer.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/caffe/layers/cudnn_lrn_layer.cpp b/src/caffe/layers/cudnn_lrn_layer.cpp index 0495b802baf..3b7985ba0bc 100644 --- a/src/caffe/layers/cudnn_lrn_layer.cpp +++ b/src/caffe/layers/cudnn_lrn_layer.cpp @@ -43,6 +43,8 @@ CuDNNLRNLayer::~CuDNNLRNLayer() { cudnnDestroyTensorDescriptor(bottom_desc_); cudnnDestroyTensorDescriptor(top_desc_); + cudnnDestroyLRNDescriptor(norm_desc_); + // destroy LRN handle cudnnDestroy(handle_); }