diff --git a/tests/dnn/.gitignore b/tests/dnn/.gitignore index d78c5c1aecb9f241c7d78ecc46f6e17abcfa3917..1fcd2410b47494d1cdbd3883d68ce8f0b4bf1a03 100644 --- a/tests/dnn/.gitignore +++ b/tests/dnn/.gitignore @@ -3,3 +3,4 @@ /dnn-layer-maximum-test /dnn-layer-pad-test /dnn-layer-mathbinary-test +/dnn-layer-mathunary-test diff --git a/tests/dnn/Makefile b/tests/dnn/Makefile index 1f96710821c592811964c63cc11c5caf7ca8a1fc..64591b7851c36c069e7c084ffeff208c811391e7 100644 --- a/tests/dnn/Makefile +++ b/tests/dnn/Makefile @@ -3,6 +3,7 @@ DNNTESTPROGS += dnn-layer-conv2d DNNTESTPROGS += dnn-layer-depth2space DNNTESTPROGS += dnn-layer-mathbinary DNNTESTPROGS += dnn-layer-maximum +DNNTESTPROGS += dnn-layer-mathunary DNNTESTOBJS := $(DNNTESTOBJS:%=$(DNNTESTSDIR)%) $(DNNTESTPROGS:%=$(DNNTESTSDIR)/%-test.o) DNNTESTPROGS := $(DNNTESTPROGS:%=$(DNNTESTSDIR)/%-test$(EXESUF)) diff --git a/tests/dnn/dnn-layer-mathunary-test.c b/tests/dnn/dnn-layer-mathunary-test.c new file mode 100644 index 0000000000000000000000000000000000000000..f032ca0684a7556dbaf137b21784cfac0df52820 --- /dev/null +++ b/tests/dnn/dnn-layer-mathunary-test.c @@ -0,0 +1,81 @@ +/* + * Copyright (c) 2020 + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include +#include +#include +#include "libavfilter/dnn/dnn_backend_native_layer_mathunary.h" +#include "libavutil/avassert.h" + +#define EPS 0.00001 + +static float get_expected(float f, DNNMathUnaryOperation op) +{ + switch (op) + { + case DMUO_ABS: + return (f >= 0) ? f : -f; + default: + av_assert0(!"not supported yet"); + return 0.f; + } +} + +static int test(DNNMathUnaryOperation op) +{ + DnnLayerMathUnaryParams params; + DnnOperand operands[2]; + int32_t input_indexes[1]; + float input[1*1*2*3] = { + -3, 2.5, 2, -2.1, 7.8, 100}; + float *output; + + params.un_op = op; + + operands[0].data = input; + operands[0].dims[0] = 1; + operands[0].dims[1] = 1; + operands[0].dims[2] = 2; + operands[0].dims[3] = 3; + operands[1].data = NULL; + + input_indexes[0] = 0; + dnn_execute_layer_math_unary(operands, input_indexes, 1, ¶ms); + + 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) { + printf("at index %d, output: %f, expected_output: %f\n", i, output[i], expected_output); + av_freep(&output); + return 1; + } + } + + av_freep(&output); + return 0; +} + +int main(int agrc, char **argv) +{ + if (test(DMUO_ABS)) + return 1; + return 0; +} diff --git a/tests/fate/dnn.mak b/tests/fate/dnn.mak index 5a8e6296a6095cc317b1f9862b83b29d1b096d1a..4a50b1638206f3762cf2a2fa39c694d3641328b6 100644 --- a/tests/fate/dnn.mak +++ b/tests/fate/dnn.mak @@ -23,6 +23,11 @@ fate-dnn-layer-maximum: $(DNNTESTSDIR)/dnn-layer-maximum-test$(EXESUF) fate-dnn-layer-maximum: CMD = run $(DNNTESTSDIR)/dnn-layer-maximum-test$(EXESUF) fate-dnn-layer-maximum: CMP = null +FATE_DNN += fate-dnn-layer-mathunary +fate-dnn-layer-mathunary: $(DNNTESTSDIR)/dnn-layer-mathunary-test$(EXESUF) +fate-dnn-layer-mathunary: CMD = run $(DNNTESTSDIR)/dnn-layer-mathunary-test$(EXESUF) +fate-dnn-layer-mathunary: CMP = null + FATE-yes += $(FATE_DNN) fate-dnn: $(FATE_DNN)