mirror of
https://github.com/xenia-project/FFmpeg.git
synced 2024-11-24 03:59:43 +00:00
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>
This commit is contained in:
parent
d294716c22
commit
dbf66b9e0e
@ -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;
|
||||
|
Loading…
Reference in New Issue
Block a user