diff --git a/dlib/test/dnn.cpp b/dlib/test/dnn.cpp index 2fbcd513d..fd5bc05c5 100644 --- 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 } {