diff options
author | morgolock <pablo.tello@arm.com> | 2020-08-20 14:51:39 +0100 |
---|---|---|
committer | Pablo Marquez <pablo.tello@arm.com> | 2020-08-25 13:17:09 +0000 |
commit | 5111264954e2d1a4d3e91d23a0869a0d7105be4c (patch) | |
tree | 57c8db6a8911378129364bc6d826a3c0ef7f9283 /arm_compute/runtime/IScheduler.h | |
parent | 96a14008af85725d067cdd8247023474581102ea (diff) | |
download | ComputeLibrary-5111264954e2d1a4d3e91d23a0869a0d7105be4c.tar.gz |
COMPMID-3661: Added multidimension support to OMP scheduler.
Change-Id: Iedacf7094896f08d7c2847c8fb99bd7153deba2c
Signed-off-by: morgolock <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3809
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Diffstat (limited to 'arm_compute/runtime/IScheduler.h')
-rw-r--r-- | arm_compute/runtime/IScheduler.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/runtime/IScheduler.h b/arm_compute/runtime/IScheduler.h index 98627538e8..309aee3bb5 100644 --- a/arm_compute/runtime/IScheduler.h +++ b/arm_compute/runtime/IScheduler.h @@ -205,6 +205,8 @@ protected: virtual void run_workloads(std::vector<Workload> &workloads) = 0; CPUInfo _cpu_info; + void schedule_common(ICPPKernel *kernel, const Hints &hints, ITensorPack &tensors); + private: unsigned int _num_threads_hint = {}; }; |