Added missing requirements check.

2016-01-23 10:37:21 -05:00 · 2016-01-23 10:37:21 -05:00 · e44b2aa266
parent 21e3e81fa0
commit e44b2aa266
1 changed files with 3 additions and 1 deletions
--- a/dlib/dnn/cudnn_dlibapi.cpp
+++ b/dlib/dnn/cudnn_dlibapi.cpp
@ -268,7 +268,9 @@ namespace dlib
                  grad.nr() == 1 &&
                  grad.nc() == 1 &&
                  gradient_input.k() == grad.k() &&
-                  gradient_input.size() > 0,"");
+                  gradient_input.size() > 0 &&
+                  is_same_object(grad,gradient_input) == false
+                  ,"");

            const float alpha = 1;
            const float beta = 0;