diff options
Diffstat (limited to 'arm_compute/runtime')
-rw-r--r-- | arm_compute/runtime/CPP/CPPScheduler.h | 1 | ||||
-rw-r--r-- | arm_compute/runtime/IScheduler.h | 10 | ||||
-rw-r--r-- | arm_compute/runtime/OMP/OMPScheduler.h | 2 | ||||
-rw-r--r-- | arm_compute/runtime/SingleThreadScheduler.h | 1 |
4 files changed, 13 insertions, 1 deletions
diff --git a/arm_compute/runtime/CPP/CPPScheduler.h b/arm_compute/runtime/CPP/CPPScheduler.h index 30bc4c8b70..17ed8310a4 100644 --- a/arm_compute/runtime/CPP/CPPScheduler.h +++ b/arm_compute/runtime/CPP/CPPScheduler.h @@ -61,6 +61,7 @@ public: */ void schedule(ICPPKernel *kernel, const Hints &hints) override; +protected: /** Will run the workloads in parallel using num_threads * * @param[in] workloads Workloads to run diff --git a/arm_compute/runtime/IScheduler.h b/arm_compute/runtime/IScheduler.h index 1f90f4ef9c..14acf04439 100644 --- a/arm_compute/runtime/IScheduler.h +++ b/arm_compute/runtime/IScheduler.h @@ -134,8 +134,9 @@ public: * @note there is no guarantee regarding the order in which the workloads will be executed or whether or not they will be executed in parallel. * * @param[in] workloads Array of workloads to run + * @param[in] tag String that can be used by profiling tools to identify the workloads run by the scheduler (Can be null). */ - virtual void run_workloads(std::vector<Workload> &workloads) = 0; + virtual void run_tagged_workloads(std::vector<Workload> &workloads, const char *tag); /** Get CPU info. * @@ -152,6 +153,13 @@ public: unsigned int num_threads_hint() const; protected: + /** Execute all the passed workloads + * + * @note there is no guarantee regarding the order in which the workloads will be executed or whether or not they will be executed in parallel. + * + * @param[in] workloads Array of workloads to run + */ + virtual void run_workloads(std::vector<Workload> &workloads) = 0; CPUInfo _cpu_info; private: diff --git a/arm_compute/runtime/OMP/OMPScheduler.h b/arm_compute/runtime/OMP/OMPScheduler.h index 4ff787630f..ff9bf052fd 100644 --- a/arm_compute/runtime/OMP/OMPScheduler.h +++ b/arm_compute/runtime/OMP/OMPScheduler.h @@ -57,6 +57,8 @@ public: * @param[in] hints Hints for the scheduler. */ void schedule(ICPPKernel *kernel, const Hints &hints) override; + +protected: /** Execute all the passed workloads * * @note there is no guarantee regarding the order in which the workloads will be executed or whether or not they will be executed in parallel. diff --git a/arm_compute/runtime/SingleThreadScheduler.h b/arm_compute/runtime/SingleThreadScheduler.h index 6924601903..7c084efeaf 100644 --- a/arm_compute/runtime/SingleThreadScheduler.h +++ b/arm_compute/runtime/SingleThreadScheduler.h @@ -54,6 +54,7 @@ public: */ void schedule(ICPPKernel *kernel, const Hints &hints) override; +protected: /** Will run the workloads sequentially and in order. * * @param[in] workloads Workloads to run |