Fixed a slow memory leak that occurred when using cuDNN and tensors were created.

ee091851 · Davis King · 84cd330c · ee091851
Commit ee091851 authored Mar 05, 2017 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 7 deletions

cudnn_dlibapi.cpp dlib/dnn/cudnn_dlibapi.cpp +5 -7

No files found.
--- a/dlib/dnn/cudnn_dlibapi.cpp
+++ b/dlib/dnn/cudnn_dlibapi.cpp
@@ -191,15 +191,13 @@ namespace dlib
            int nc 
        )
        {
-            if (n == 0 || nr == 0 || nc == 0 || k == 0)
+            if (handle)
            {
-                if (handle)
-                {
-                    cudnnDestroyTensorDescriptor((cudnnTensorDescriptor_t)handle);
-                    handle = nullptr;
-                }
+                cudnnDestroyTensorDescriptor((cudnnTensorDescriptor_t)handle);
+                handle = nullptr;
            }
-            else
+
+            if (n != 0 && nr != 0 && nc != 0 && k != 0)
            {
                cudnnTensorDescriptor_t h;
                CHECK_CUDNN(cudnnCreateTensorDescriptor(&h));