Added missing assert check

20d46fc5 · Davis King · a29086bf · 20d46fc5
Commit 20d46fc5 authored Dec 09, 2015 by Davis King
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

cudnn_dlibapi.cpp dlib/dnn/cudnn_dlibapi.cpp +2 -1

No files found.
--- a/dlib/dnn/cudnn_dlibapi.cpp
+++ b/dlib/dnn/cudnn_dlibapi.cpp
@@ -196,7 +196,8 @@ namespace dlib
                  (dest.num_samples()==src.num_samples() || src.num_samples()==1) &&
                  (dest.nr()==src.nr() || src.nr()==1) &&
                  (dest.nc()==src.nc() || src.nc()==1) &&
-                  (dest.k()==src.k()   || src.k()==1), "");
+                  (dest.k()==src.k()   || src.k()==1) &&
+                  is_same_object(src,dest) == false , "");

            CHECK_CUDNN(cudnnAddTensor_v3(context(),
                                    &alpha,