Skip to content

D
dlib
Commit 16583730 authored by Davis King's avatar Davis King
Browse files
Options

fixed bug in tests

parent 24830e8e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
dlib/test/dnn.cpp
View file @ 16583730
...@@ -458,6 +458,7 @@ namespace ...@@ -458,6 +458,7 @@ namespace
memcpy(A, truth); memcpy(A, truth);
DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5); DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5);
#ifdef DLIB_USE_CUDA
A = 4; A = 4;
A.device(); A.device();
B.host(); B.host();
...@@ -481,6 +482,7 @@ namespace ...@@ -481,6 +482,7 @@ namespace
B.device(); B.device();
memcpy(A, truth); memcpy(A, truth);
DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5); DLIB_TEST(max(abs(mat(A)- mat(truth))) < 1e-5);
#endif
} }
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment