fixed bug in tests

16583730 · Davis King · 24830e8e · 16583730
Commit 16583730 authored Apr 25, 2016 by Davis King
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

dnn.cpp dlib/test/dnn.cpp +2 -0

No files found.
--- a/dlib/test/dnn.cpp
+++ b/dlib/test/dnn.cpp
@@ -458,6 +458,7 @@ namespace
            memcpy(A, truth);
            DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5);

+#ifdef DLIB_USE_CUDA
            A = 4;
            A.device();
            B.host();
@@ -481,6 +482,7 @@ namespace
            B.device();
            memcpy(A, truth);
            DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5);
+#endif
        }

        {