Gitiles
Code Review Sign In
review.mlplatform.org / ml / ComputeLibrary / 8d5d5888427a0f9c721fa745f0ea457dcf298d03 / . / tests / benchmark / GLES_COMPUTE / ConvolutionLayer.cpp
blob: c79e0156cff7353ed6f0da353b0d19e749abeb17 [file] [log] [blame]
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]1/*
2 * Copyright (c) 2017-2018 ARM Limited.
3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24#include "arm_compute/core/TensorShape.h"
25#include "arm_compute/core/Types.h"
26#include "arm_compute/runtime/GLES_COMPUTE/GCTensor.h"
27#include "arm_compute/runtime/GLES_COMPUTE/GCTensorAllocator.h"
28#include "arm_compute/runtime/GLES_COMPUTE/functions/GCConvolutionLayer.h"
29#include "tests/GLES_COMPUTE/GCAccessor.h"
30#include "tests/benchmark/fixtures/ConvolutionLayerFixture.h"
31#include "tests/datasets/system_tests/alexnet/AlexNetConvolutionLayerDataset.h"
32#include "tests/datasets/system_tests/googlenet/inceptionv1/GoogLeNetInceptionV1ConvolutionLayerDataset.h"
33#include "tests/datasets/system_tests/googlenet/inceptionv4/GoogLeNetInceptionV4ConvolutionLayerDataset.h"
34#include "tests/datasets/system_tests/lenet5/LeNet5ConvolutionLayerDataset.h"
35#include "tests/datasets/system_tests/squeezenet/SqueezeNetConvolutionLayerDataset.h"
36#include "tests/datasets/system_tests/vgg/vgg16/VGG16ConvolutionLayerDataset.h"
37#include "tests/datasets/system_tests/yolo/v2/YOLOV2ConvolutionLayerDataset.h"
38#include "tests/framework/Macros.h"
39#include "tests/framework/datasets/Datasets.h"
40#include "utils/TypePrinter.h"
41
42namespace arm_compute
43{
44namespace test
45{
Michalis Spyrou724079d2017-12-15 11:37:37 +0000[diff] [blame]46namespace benchmark
47{
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]48namespace
49{
50const auto data_types = framework::dataset::make("DataType", { DataType::F16 });
51} // namespace
52
53using GCConvolutionLayerFixture = ConvolutionLayerFixture<GCTensor, GCConvolutionLayer, GCAccessor>;
54
55TEST_SUITE(GC)
56
57REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]58 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::AlexNetConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
59 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]60 data_types),
61 framework::dataset::make("Batches", 1)));
62
63REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5ConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]64 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::LeNet5ConvolutionLayerDataset(), framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]65 data_types),
66 framework::dataset::make("Batches", 1)));
67
68REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1ConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]69 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1ConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
70 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]71 data_types),
72 framework::dataset::make("Batches", 1)));
73
74REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4ConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]75 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4ConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
76 ActivationLayerInfo())),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]77 data_types),
78 framework::dataset::make("Batches", 1)));
79
80REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::ALL,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]81 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
82 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]83 data_types),
84 framework::dataset::make("Batches", 1)));
85
86TEST_SUITE(NIGHTLY)
87REGISTER_FIXTURE_DATA_TEST_CASE(AlexNetConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]88 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::AlexNetConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
89 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]90 data_types),
91 framework::dataset::make("Batches", { 4, 8 })));
92
93REGISTER_FIXTURE_DATA_TEST_CASE(LeNet5ConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]94 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::LeNet5ConvolutionLayerDataset(), framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]95 data_types),
96 framework::dataset::make("Batches", { 4, 8 })));
97
98REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV1ConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]99 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV1ConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
100 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]101 data_types),
102 framework::dataset::make("Batches", { 4, 8 })));
103
104REGISTER_FIXTURE_DATA_TEST_CASE(GoogLeNetInceptionV4ConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]105 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::GoogLeNetInceptionV4ConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
106 ActivationLayerInfo())),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]107 data_types),
108 framework::dataset::make("Batches", { 4, 8 })));
109
110REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]111 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
112 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]113 data_types),
114 framework::dataset::make("Batches", { 4, 8 })));
115
116// 8 batches use about 1.8GB of memory which is too much for most devices!
117REGISTER_FIXTURE_DATA_TEST_CASE(VGG16ConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]118 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::VGG16ConvolutionLayerDataset(), framework::dataset::make("ActivationInfo",
119 ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU))),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]120 data_types),
121 framework::dataset::make("Batches", { 1, 4 })));
122
123REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2ConvolutionLayer, GCConvolutionLayerFixture, framework::DatasetMode::NIGHTLY,
Isabella Gottardi3f217ec2018-02-12 14:59:19 +0000[diff] [blame]124 framework::dataset::combine(framework::dataset::combine(framework::dataset::combine(datasets::YOLOV2ConvolutionLayerDataset(), framework::dataset::make("ActivationInfo", ActivationLayerInfo())),
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]125 data_types),
126 framework::dataset::make("Batches", { 1, 4, 8 })));
127TEST_SUITE_END()
128TEST_SUITE_END()
Michalis Spyrou724079d2017-12-15 11:37:37 +0000[diff] [blame]129} // namespace benchmark
Stephen Lie855c232018-01-04 14:13:22 +0800[diff] [blame]130} // namespace test
131} // namespace arm_compute