fixed bug in tests

2016-04-25 07:01:36 -04:00 · 2016-04-25 07:01:36 -04:00 · 16583730d7
parent 24830e8e2c
commit 16583730d7
1 changed files with 2 additions and 0 deletions
--- 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
        }

        {