mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

conv2d destry

Browse Source
This commit is contained in:
Francesco Gatti
2017-06-28 15:29:14 +00:00
parent b9f894ca75
commit cfa43a8580
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Conv2d.cpp
+5 -1
View File
@@ -16,9 +16,9 @@ Conv2d::Conv2d( Network *net, dataDim_t in_dim, int out_ch,
this->strideH = strideH; this->strideH = strideH;
this->strideW = strideW; this->strideW = strideW;
checkCUDNN( cudnnCreateTensorDescriptor(&biasTensorDesc) );
checkCUDNN( cudnnCreateFilterDescriptor(&filterDesc) ); checkCUDNN( cudnnCreateFilterDescriptor(&filterDesc) );
checkCUDNN( cudnnCreateConvolutionDescriptor(&convDesc) ); checkCUDNN( cudnnCreateConvolutionDescriptor(&convDesc) );
checkCUDNN( cudnnCreateTensorDescriptor(&biasTensorDesc) );
int n = input_dim.n; int n = input_dim.n;
int c = input_dim.c; int c = input_dim.c;
@@ -79,6 +79,10 @@ Conv2d::Conv2d( Network *net, dataDim_t in_dim, int out_ch,
Conv2d::~Conv2d() { Conv2d::~Conv2d() {
checkCUDNN( cudnnDestroyFilterDescriptor(filterDesc) );
checkCUDNN( cudnnDestroyConvolutionDescriptor(convDesc) );
checkCUDNN( cudnnDestroyTensorDescriptor(biasTensorDesc) );
if (ws_sizeInBytes!=0) if (ws_sizeInBytes!=0)
checkCuda( cudaFree(workSpace) ); checkCuda( cudaFree(workSpace) );