[FFmpeg-devel] [PATCH] tests/dnn/mathunary: fix the issue of NAN
Ting Fu
ting.fu at intel.com
Thu Jul 2 16:51:24 EEST 2020
When one of output[i] & expected_output is NAN, the unit test will always pass.
Signed-off-by: Ting Fu <ting.fu at intel.com>
---
tests/dnn/dnn-layer-mathunary-test.c | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/dnn/dnn-layer-mathunary-test.c b/tests/dnn/dnn-layer-mathunary-test.c
index bf77c44bbe..f251447771 100644
--- a/tests/dnn/dnn-layer-mathunary-test.c
+++ b/tests/dnn/dnn-layer-mathunary-test.c
@@ -74,7 +74,8 @@ 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) {
+ if ((isnan(output[i]) ^ isnan(expected_output)) ||
+ fabs(output[i] - expected_output) > EPS) {
printf("at index %d, output: %f, expected_output: %f\n", i, output[i], expected_output);
av_freep(&output);
return 1;
--
2.17.1
More information about the ffmpeg-devel
mailing list