From d771689903e336fa4e3293c5f204520cb1d799f8 Mon Sep 17 00:00:00 2001 From: Tomohiro Ubukata Date: Sun, 1 Mar 2020 07:07:52 +0000 Subject: [PATCH] Fix CuDNNLCNLayer Resource leak --- src/caffe/layers/cudnn_lcn_layer.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/caffe/layers/cudnn_lcn_layer.cpp b/src/caffe/layers/cudnn_lcn_layer.cpp index 9c09bf26b4d..0dc04db1e51 100644 --- a/src/caffe/layers/cudnn_lcn_layer.cpp +++ b/src/caffe/layers/cudnn_lcn_layer.cpp @@ -59,6 +59,8 @@ CuDNNLCNLayer::~CuDNNLCNLayer() { cudnnDestroyTensorDescriptor(bottom_desc_); cudnnDestroyTensorDescriptor(top_desc_); + cudnnDestroyLRNDescriptor(norm_desc_); + // destroy LRN handle cudnnDestroy(handle_);