diff options
Diffstat (limited to 'framework')
-rw-r--r-- | framework/Framework.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/framework/Framework.cpp b/framework/Framework.cpp index 7708537fd6..e3ada039ae 100644 --- a/framework/Framework.cpp +++ b/framework/Framework.cpp @@ -28,6 +28,7 @@ #ifdef ARM_COMPUTE_CL #include "arm_compute/core/CL/OpenCL.h" +#include "arm_compute/runtime/CL/CLScheduler.h" #endif /* ARM_COMPUTE_CL */ #include <chrono> @@ -226,6 +227,12 @@ void Framework::run_test(TestCaseFactory &test_factory) { profiler.start(); test_case->do_run(); +#ifdef ARM_COMPUTE_CL + if(opencl_is_available()) + { + CLScheduler::get().sync(); + } +#endif /* ARM_COMPUTE_CL */ profiler.stop(); } |