diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt index 42a9438bc..66fb8ccca 100644 --- a/tests/CMakeLists.txt +++ b/tests/CMakeLists.txt @@ -47,3 +47,4 @@ ncnn_add_layer_test(Slice) ncnn_add_layer_test(Softmax) ncnn_add_layer_test(UnaryOp) ncnn_add_layer_test(TanH) +ncnn_add_layer_test(Clip) diff --git a/tests/test_clip.cpp b/tests/test_clip.cpp new file mode 100644 index 000000000..88253b959 --- /dev/null +++ b/tests/test_clip.cpp @@ -0,0 +1,78 @@ +// Tencent is pleased to support the open source community by making ncnn available. +// +// Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved. +// +// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except +// in compliance with the License. You may obtain a copy of the License at +// +// https://opensource.org/licenses/BSD-3-Clause +// +// Unless required by applicable law or agreed to in writing, software distributed +// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR +// CONDITIONS OF ANY KIND, either express or implied. See the License for the +// specific language governing permissions and limitations under the License. + +#include "testutil.h" + +#include "layer/clip.h" + +static int test_clip(const ncnn::Mat &a, float min, float max) +{ + ncnn::ParamDict pd; + pd.set(0, min); + pd.set(1, max); + + std::vector weights(0); + + ncnn::Option opt; + opt.num_threads = 1; + opt.use_vulkan_compute = true; + opt.use_fp16_packed = false; + opt.use_fp16_storage = false; + opt.use_fp16_arithmetic = false; + opt.use_int8_storage = false; + opt.use_int8_arithmetic = false; + + int ret = test_layer("Clip", pd, weights, opt, a); + if (ret != 0) + { + fprintf(stderr, "test_clip failed min=%f max=%f\n", min, max); + } + + return ret; +} + +static int test_clip_0() +{ + return 0 + || test_clip(RandomMat(6, 7, 16), -1.f, 1.f) + || test_clip(RandomMat(3, 5, 13), -1.f, 1.f) + ; +} + +static int test_clip_1() +{ + return 0 + || test_clip(RandomMat(6, 16), -1.f, 1.f) + || test_clip(RandomMat(7, 15), -1.f, 1.f) + ; +} + +static int test_clip_2() +{ + return 0 + || test_clip(RandomMat(128), -1.f, 1.f) + || test_clip(RandomMat(127), -1.f, 1.f) + ; +} + +int main() +{ + SRAND(7767517); + + return 0 + || test_clip_0() + || test_clip_1() + || test_clip_2() + ; +}