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

memory release

Browse Source
This commit is contained in:
Francesco Gatti
2020-06-02 12:43:06 +02:00
parent 8c79b75227
commit 4f907dde40
7 changed files with 20 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/Layer.cpp
+5
View File
@@ -24,6 +24,11 @@ Layer::~Layer() {
checkCUDNN( cudnnDestroyTensorDescriptor(srcTensorDesc) );
checkCUDNN( cudnnDestroyTensorDescriptor(dstTensorDesc) );
if(dstData != nullptr) {
cudaFree(dstData);
dstData = nullptr;
}
}
}}