When one of output[i] & expected_output is NAN, the unit test will always pass.

Signed-off-by: Ting Fu <ting...@intel.com>
---
 tests/dnn/dnn-layer-mathunary-test.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/dnn/dnn-layer-mathunary-test.c 
b/tests/dnn/dnn-layer-mathunary-test.c
index 683e623d95..5afc5c157e 100644
--- a/tests/dnn/dnn-layer-mathunary-test.c
+++ b/tests/dnn/dnn-layer-mathunary-test.c
@@ -86,7 +86,10 @@ static int test(DNNMathUnaryOperation op)
     output = operands[1].data;
     for (int i = 0; i < sizeof(input) / sizeof(float); ++i) {
         float expected_output = get_expected(input[i], op);
-        if(fabs(output[i] - expected_output) > EPS) {
+        int output_nan = isnan(output[i]);
+        int expected_nan = isnan(expected_output);
+        if ((!output_nan && !expected_nan && fabs(output[i] - expected_output) 
> EPS) ||
+            (output_nan && !expected_nan) || (!output_nan && expected_nan)) {
             printf("at index %d, output: %f, expected_output: %f\n", i, 
output[i], expected_output);
             av_freep(&output);
             return 1;
-- 
2.17.1

_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".

Reply via email to