diff options
Diffstat (limited to 'tests/benchmark/fixtures/ConvolutionLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/ConvolutionLayerFixture.h | 15 |
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() |