diff options
author | zhenglin <zhenglin.li@arm.com> | 2017-12-05 11:30:51 +0800 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:17 +0000 |
commit | 923241eb998ad031f4cce7b12d8c24a0b6c80be8 (patch) | |
tree | 9cec664884d324c6a5cd8ebc5816071380417de2 /tests/benchmark/fixtures/BatchNormalizationLayerFixture.h | |
parent | 0bc5a258903bbbf8623b6e1c4253d958551293ed (diff) | |
download | ComputeLibrary-923241eb998ad031f4cce7b12d8c24a0b6c80be8.tar.gz |
APPBROWSER-314: Performance optimazation for BatchNormalizationLayer
Change-Id: Ie3ad9abb64e90720609bb6e67662eaf9dd4f3689
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/111826
Reviewed-by: Joel Liang <joel.liang@arm.com>
Tested-by: BSG Visual Compute Jenkins server to access repositories on http://mpd-gerrit.cambridge.arm.com <bsgcomp@arm.com>
(cherry picked from commit 02c1fa663926cc4fcd1995d4d18d7528e0c85d94)
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/111834
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/benchmark/fixtures/BatchNormalizationLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/BatchNormalizationLayerFixture.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/benchmark/fixtures/BatchNormalizationLayerFixture.h b/tests/benchmark/fixtures/BatchNormalizationLayerFixture.h index 79dbc76300..55411a44d1 100644 --- a/tests/benchmark/fixtures/BatchNormalizationLayerFixture.h +++ b/tests/benchmark/fixtures/BatchNormalizationLayerFixture.h @@ -29,6 +29,12 @@ #include "tests/Globals.h" #include "tests/Utils.h" #include "tests/framework/Fixture.h" +#ifdef ARM_COMPUTE_GC +#include "arm_compute/runtime/GLES_COMPUTE/GCScheduler.h" +#include "tests/GLES_COMPUTE/Helper.h" + +using namespace arm_compute::test::gles_compute; +#endif /* ARM_COMPUTE_GC */ namespace arm_compute { @@ -76,6 +82,12 @@ public: void run() { batch_norm_layer.run(); +#ifdef ARM_COMPUTE_GC + if(opengles31_is_available() && std::is_same<typename std::decay<TensorType>::type, arm_compute::GCTensor>::value) + { + force_sync_tensor(dst); + } +#endif /* ARM_COMPUTE_GC */ } void teardown() |