From patchwork Mon Mar 23 04:08:23 2020 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: "Guo, Yejun" X-Patchwork-Id: 18363 Return-Path: X-Original-To: patchwork@ffaux-bg.ffmpeg.org Delivered-To: patchwork@ffaux-bg.ffmpeg.org Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by ffaux.localdomain (Postfix) with ESMTP id 7322E44B13D for ; Mon, 23 Mar 2020 06:09:26 +0200 (EET) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 61B5868B762; Mon, 23 Mar 2020 06:09:26 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mga09.intel.com (mga09.intel.com [134.134.136.24]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 3B17B68B755 for ; Mon, 23 Mar 2020 06:09:19 +0200 (EET) IronPort-SDR: viiMVlOiXBb1HCBzPI/86jQ6LiJIdsM6xOh5sYDSN5yRzvgIOAUKB4ZlaiGo8ddJgetreq0xVF BJNee1pdNIug== X-Amp-Result: SKIPPED(no attachment in message) X-Amp-File-Uploaded: False Received: from orsmga003.jf.intel.com ([10.7.209.27]) by orsmga102.jf.intel.com with ESMTP/TLS/ECDHE-RSA-AES256-GCM-SHA384; 22 Mar 2020 21:09:17 -0700 IronPort-SDR: 8O+Qz2+xc+JsNE0QRGM+yGlpJ/G/mjonUv1A0pdDkK/aPdHC8Y8lZXHCMiT4LE0OaMUmOu7RhD uF+qh0r8WYtg== X-ExtLoop1: 1 X-IronPort-AV: E=Sophos;i="5.72,295,1580803200"; d="scan'208";a="246073131" Received: from yguo18-skl-u1604.sh.intel.com ([10.239.159.53]) by orsmga003.jf.intel.com with ESMTP; 22 Mar 2020 21:09:16 -0700 From: "Guo, Yejun" To: ffmpeg-devel@ffmpeg.org Date: Mon, 23 Mar 2020 12:08:23 +0800 Message-Id: <1584936503-9191-1-git-send-email-yejun.guo@intel.com> X-Mailer: git-send-email 2.7.4 Subject: [FFmpeg-devel] [PATCH 3/4] dnn-layer-mathbinary-test: add unit test for subtraction X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: yejun.guo@intel.com MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Signed-off-by: Guo, Yejun --- tests/dnn/.gitignore | 1 + tests/dnn/Makefile | 1 + tests/dnn/dnn-layer-mathbinary-test.c | 173 ++++++++++++++++++++++++++++++++++ tests/fate/dnn.mak | 5 + 4 files changed, 180 insertions(+) create mode 100644 tests/dnn/dnn-layer-mathbinary-test.c diff --git a/tests/dnn/.gitignore b/tests/dnn/.gitignore index 5eedaaa..d78c5c1 100644 --- a/tests/dnn/.gitignore +++ b/tests/dnn/.gitignore @@ -2,3 +2,4 @@ /dnn-layer-depth2space-test /dnn-layer-maximum-test /dnn-layer-pad-test +/dnn-layer-mathbinary-test diff --git a/tests/dnn/Makefile b/tests/dnn/Makefile index e1bfe3f..1f96710 100644 --- a/tests/dnn/Makefile +++ b/tests/dnn/Makefile @@ -1,6 +1,7 @@ DNNTESTPROGS += dnn-layer-pad DNNTESTPROGS += dnn-layer-conv2d DNNTESTPROGS += dnn-layer-depth2space +DNNTESTPROGS += dnn-layer-mathbinary DNNTESTPROGS += dnn-layer-maximum DNNTESTOBJS := $(DNNTESTOBJS:%=$(DNNTESTSDIR)%) $(DNNTESTPROGS:%=$(DNNTESTSDIR)/%-test.o) diff --git a/tests/dnn/dnn-layer-mathbinary-test.c b/tests/dnn/dnn-layer-mathbinary-test.c new file mode 100644 index 0000000..1243784 --- /dev/null +++ b/tests/dnn/dnn-layer-mathbinary-test.c @@ -0,0 +1,173 @@ +/* + * 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_mathbinary.h" + +#define EPSON 0.00001 + +static int test_sub_broadcast_input0(void) +{ + DnnLayerMathBinaryParams 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.bin_op = DMBO_SUB; + params.input0_broadcast = 1; + params.input1_broadcast = 0; + params.v = 7.28; + + 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_binary(operands, input_indexes, 1, ¶ms); + + output = operands[1].data; + for (int i = 0; i < sizeof(input) / sizeof(float); i++) { + float expected_output = params.v - input[i]; + if (fabs(output[i] - expected_output) > EPSON) { + 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; +} + +static int test_sub_broadcast_input1(void) +{ + DnnLayerMathBinaryParams 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.bin_op = DMBO_SUB; + params.input0_broadcast = 0; + params.input1_broadcast = 1; + params.v = 7.28; + + 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_binary(operands, input_indexes, 1, ¶ms); + + output = operands[1].data; + for (int i = 0; i < sizeof(input) / sizeof(float); i++) { + float expected_output = input[i] - params.v; + if (fabs(output[i] - expected_output) > EPSON) { + 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; +} + +static int test_sub_no_broadcast(void) +{ + DnnLayerMathBinaryParams params; + DnnOperand operands[3]; + int32_t input_indexes[2]; + float input0[1*1*2*3] = { + -3, 2.5, 2, -2.1, 7.8, 100 + }; + float input1[1*1*2*3] = { + -1, 2, 3, -21, 8, 10.0 + }; + float *output; + + params.bin_op = DMBO_SUB; + params.input0_broadcast = 0; + params.input1_broadcast = 0; + + operands[0].data = input0; + operands[0].dims[0] = 1; + operands[0].dims[1] = 1; + operands[0].dims[2] = 2; + operands[0].dims[3] = 3; + operands[1].data = input1; + operands[1].dims[0] = 1; + operands[1].dims[1] = 1; + operands[1].dims[2] = 2; + operands[1].dims[3] = 3; + operands[2].data = NULL; + + input_indexes[0] = 0; + input_indexes[1] = 1; + dnn_execute_layer_math_binary(operands, input_indexes, 2, ¶ms); + + output = operands[2].data; + for (int i = 0; i < sizeof(input0) / sizeof(float); i++) { + float expected_output = input0[i] - input1[i]; + if (fabs(output[i] - expected_output) > EPSON) { + 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; +} + +static int test_sub(void) +{ + if (test_sub_broadcast_input0()) + return 1; + + if (test_sub_broadcast_input1()) + return 1; + + if (test_sub_no_broadcast()) + return 1; + + return 0; +} + +int main(int argc, char **argv) +{ + if (test_sub()) + return 1; + + return 0; +} diff --git a/tests/fate/dnn.mak b/tests/fate/dnn.mak index ec60b07..5a8e629 100644 --- a/tests/fate/dnn.mak +++ b/tests/fate/dnn.mak @@ -13,6 +13,11 @@ fate-dnn-layer-depth2space: $(DNNTESTSDIR)/dnn-layer-depth2space-test$(EXESUF) fate-dnn-layer-depth2space: CMD = run $(DNNTESTSDIR)/dnn-layer-depth2space-test$(EXESUF) fate-dnn-layer-depth2space: CMP = null +FATE_DNN += fate-dnn-layer-mathbinary +fate-dnn-layer-mathbinary: $(DNNTESTSDIR)/dnn-layer-mathbinary-test$(EXESUF) +fate-dnn-layer-mathbinary: CMD = run $(DNNTESTSDIR)/dnn-layer-mathbinary-test$(EXESUF) +fate-dnn-layer-mathbinary: CMP = null + FATE_DNN += fate-dnn-layer-maximum fate-dnn-layer-maximum: $(DNNTESTSDIR)/dnn-layer-maximum-test$(EXESUF) fate-dnn-layer-maximum: CMD = run $(DNNTESTSDIR)/dnn-layer-maximum-test$(EXESUF)