dnn-layer-mathbinary-test: add unit test for divide

Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
2020-04-11 13:50:32 +08:00 · 2020-04-11 13:50:32 +08:00 · 2e38c63630
parent 8ce9d88f93
commit 2e38c63630
1 changed files with 5 additions and 0 deletions
--- a/tests/dnn/dnn-layer-mathbinary-test.c
+++ b/tests/dnn/dnn-layer-mathbinary-test.c
@ -36,6 +36,8 @@ static float get_expected(float f1, float f2, DNNMathBinaryOperation op)
        return f1 + f2;
    case DMBO_MUL:
        return f1 * f2;
+    case DMBO_REALDIV:
+        return f1 / f2;
    default:
        av_assert0(!"not supported yet");
        return 0.f;
@ -195,5 +197,8 @@ int main(int argc, char **argv)
    if (test(DMBO_MUL))
        return 1;

+    if (test(DMBO_REALDIV))
+        return 1;
+
    return 0;
 }