aboutsummaryrefslogtreecommitdiff
path: root/tests/benchmark/fixtures/ConvolutionLayerFixture.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/benchmark/fixtures/ConvolutionLayerFixture.h')
-rw-r--r--tests/benchmark/fixtures/ConvolutionLayerFixture.h15
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/benchmark/fixtures/ConvolutionLayerFixture.h b/tests/benchmark/fixtures/ConvolutionLayerFixture.h
index fd508d4500..09e6cbfaf8 100644
--- a/tests/benchmark/fixtures/ConvolutionLayerFixture.h
+++ b/tests/benchmark/fixtures/ConvolutionLayerFixture.h
@@ -30,6 +30,13 @@
#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
{
namespace test
@@ -71,6 +78,14 @@ public:
void run()
{
conv_layer.run();
+#ifdef ARM_COMPUTE_GC
+ if(opengles31_is_available() && std::is_same<typename std::decay<TensorType>::type, arm_compute::GCTensor>::value)
+ {
+ GCScheduler::get().sync();
+ force_sync_tensor(src);
+ force_sync_tensor(dst);
+ }
+#endif /* ARM_COMPUTE_GC */
}
void teardown()