diff options
| author | Ting Fu <ting.fu@intel.com> | 2020-07-08 14:09:51 +0800 |
|---|---|---|
| committer | Guo, Yejun <yejun.guo@intel.com> | 2020-07-09 09:34:44 +0800 |
| commit | dbf66b9e0edf8869da5187921b96ab3290d17eb8 (patch) | |
| tree | 46a2e4b7f402aa2b82f855f515a964ec4f10cea6 /tests | |
| parent | d294716c22c827d889dde235a8e9fcd4dd6466c9 (diff) | |
| download | ffmpeg-dbf66b9e0edf8869da5187921b96ab3290d17eb8.tar.gz | |
tests/dnn/mathunary: fix the issue of NAN
When one of output[i] & expected_output is NAN, the unit test will always pass.
Signed-off-by: Ting Fu <ting.fu@intel.com>
Reviewed-by: Guo, Yejun <yejun.guo@intel.com>
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/dnn/dnn-layer-mathunary-test.c | 5 |
1 files changed, 4 insertions, 1 deletions
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; |