From cfa43a8580ae3d1b3b13bcbf299ffa1a73e79465 Mon Sep 17 00:00:00 2001 From: Francesco Gatti Date: Wed, 28 Jun 2017 15:29:14 +0000 Subject: [PATCH] conv2d destry --- src/Conv2d.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/Conv2d.cpp b/src/Conv2d.cpp index 0853788..677c395 100644 --- a/src/Conv2d.cpp +++ b/src/Conv2d.cpp @@ -16,9 +16,9 @@ Conv2d::Conv2d( Network *net, dataDim_t in_dim, int out_ch, this->strideH = strideH; this->strideW = strideW; - checkCUDNN( cudnnCreateTensorDescriptor(&biasTensorDesc) ); checkCUDNN( cudnnCreateFilterDescriptor(&filterDesc) ); checkCUDNN( cudnnCreateConvolutionDescriptor(&convDesc) ); + checkCUDNN( cudnnCreateTensorDescriptor(&biasTensorDesc) ); int n = input_dim.n; int c = input_dim.c; @@ -79,6 +79,10 @@ Conv2d::Conv2d( Network *net, dataDim_t in_dim, int out_ch, Conv2d::~Conv2d() { + checkCUDNN( cudnnDestroyFilterDescriptor(filterDesc) ); + checkCUDNN( cudnnDestroyConvolutionDescriptor(convDesc) ); + checkCUDNN( cudnnDestroyTensorDescriptor(biasTensorDesc) ); + if (ws_sizeInBytes!=0) checkCuda( cudaFree(workSpace) );