diff options
author | Moritz Pflanzer <moritz.pflanzer@arm.com> | 2017-09-08 16:07:10 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | 70e9bc21682f4eaedaceb632f594f588cb2c91fc (patch) | |
tree | 7d94f73004bc4638a84c3f8480736d23fab99754 | |
parent | ba0f8dd10a2d99e167c3ef0e0335765d04f62f10 (diff) | |
download | ComputeLibrary-70e9bc21682f4eaedaceb632f594f588cb2c91fc.tar.gz |
COMPMID-417: Remove 8 batches for NEON VGG convolution layer
Change-Id: I7f613de30aa09797b0035f9c9241fc2cfe076871
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/87099
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
-rw-r--r-- | tests/benchmark/NEON/ConvolutionLayer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/benchmark/NEON/ConvolutionLayer.cpp b/tests/benchmark/NEON/ConvolutionLayer.cpp index ba682825f1..6d2857d621 100644 --- a/tests/benchmark/NEON/ConvolutionLayer.cpp +++ b/tests/benchmark/NEON/ConvolutionLayer.cpp @@ -97,9 +97,10 @@ REGISTER_FIXTURE_DATA_TEST_CASE(SqueezeNetConvolutionLayer, NEConvolutionLayerFi framework::dataset::combine(framework::dataset::combine(datasets::SqueezeNetConvolutionLayerDataset(), data_types), framework::dataset::make("Batches", { 4, 8 }))); +// 8 batches use about 2GB of memory which is too much for most devices! REGISTER_FIXTURE_DATA_TEST_CASE(VGG16ConvolutionLayer, NEConvolutionLayerFixture, framework::DatasetMode::NIGHTLY, framework::dataset::combine(framework::dataset::combine(datasets::VGG16ConvolutionLayerDataset(), data_types), - framework::dataset::make("Batches", { 1, 4, 8 }))); + framework::dataset::make("Batches", { 1, 4 }))); REGISTER_FIXTURE_DATA_TEST_CASE(YOLOV2ConvolutionLayer, NEConvolutionLayerFixture, framework::DatasetMode::NIGHTLY, framework::dataset::combine(framework::dataset::combine(datasets::YOLOV2ConvolutionLayerDataset(), data_types), |