diff options
Diffstat (limited to 'tests/framework/instruments/OpenCLTimer.h')
| -rw-r--r-- | tests/framework/instruments/OpenCLTimer.h | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/tests/framework/instruments/OpenCLTimer.h b/tests/framework/instruments/OpenCLTimer.h index 44578782ed..059f4493f9 100644 --- a/tests/framework/instruments/OpenCLTimer.h +++ b/tests/framework/instruments/OpenCLTimer.h @@ -30,6 +30,8 @@ #include "arm_compute/core/CL/OpenCL.h" #endif /* ARM_COMPUTE_CL */ +#include "arm_compute/graph/Workload.h" + #include <list> namespace arm_compute @@ -48,17 +50,22 @@ public: */ OpenCLTimer(ScaleFactor scale_factor); std::string id() const override; + void test_start() override; void start() override; - void stop() override; + void test_stop() override; MeasurementsMap measurements() const override; + +private: #ifdef ARM_COMPUTE_CL struct kernel_info { cl::Event event{}; /**< OpenCL event associated to the kernel enqueue */ std::string name{}; /**< OpenCL Kernel name */ }; - std::list<kernel_info> kernels{}; - std::function<decltype(clEnqueueNDRangeKernel)> real_function; + std::list<kernel_info> _kernels; + std::function<decltype(clEnqueueNDRangeKernel)> _real_function; + std::function<decltype(graph::execute_task)> _real_graph_function; + std::string _prefix; #endif /* ARM_COMPUTE_CL */ private: |