From 5c2fb3f34462632b99331e2cc2d964c99fc1782b Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 1 May 2018 15:26:20 +0100 Subject: COMPMID-997: Add support for node's name in GraphAPI. Change-Id: I0ca02e42807c1ad9afeffb7202a3556feb11442f Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/129701 Tested-by: Jenkins Reviewed-by: Anthony Barbier Reviewed-by: Georgios Pinitas --- tests/framework/instruments/SchedulerTimer.h | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) (limited to 'tests/framework/instruments/SchedulerTimer.h') diff --git a/tests/framework/instruments/SchedulerTimer.h b/tests/framework/instruments/SchedulerTimer.h index ec282cc905..55d5f25b75 100644 --- a/tests/framework/instruments/SchedulerTimer.h +++ b/tests/framework/instruments/SchedulerTimer.h @@ -25,7 +25,9 @@ #define ARM_COMPUTE_TEST_SCHEDULER_TIMER #include "Instrument.h" +#include "arm_compute/graph/Workload.h" #include "arm_compute/runtime/Scheduler.h" + #include namespace arm_compute @@ -50,8 +52,9 @@ public: SchedulerTimer &operator=(const SchedulerTimer &) = delete; std::string id() const override; + void test_start() override; void start() override; - void stop() override; + void test_stop() override; Instrument::MeasurementsMap measurements() const override; /** Kernel information */ @@ -59,13 +62,16 @@ public: { Instrument::MeasurementsMap measurements{}; /**< Time it took the kernel to run */ std::string name{}; /**< Kernel name */ + std::string prefix{}; /**< Kernel prefix */ }; private: - std::list _kernels; - IScheduler *_real_scheduler; - Scheduler::Type _real_scheduler_type; - ScaleFactor _scale_factor; + std::list _kernels; + IScheduler *_real_scheduler; + Scheduler::Type _real_scheduler_type; + std::function _real_graph_function; + ScaleFactor _scale_factor; + std::shared_ptr _interceptor; }; } // namespace framework } // namespace test -- cgit v1.2.1