From 5111264954e2d1a4d3e91d23a0869a0d7105be4c Mon Sep 17 00:00:00 2001 From: morgolock Date: Thu, 20 Aug 2020 14:51:39 +0100 Subject: COMPMID-3661: Added multidimension support to OMP scheduler. Change-Id: Iedacf7094896f08d7c2847c8fb99bd7153deba2c Signed-off-by: morgolock Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3809 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Sang-Hoon Park --- arm_compute/runtime/IScheduler.h | 2 ++ 1 file changed, 2 insertions(+) (limited to 'arm_compute/runtime/IScheduler.h') 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 &workloads) = 0; CPUInfo _cpu_info; + void schedule_common(ICPPKernel *kernel, const Hints &hints, ITensorPack &tensors); + private: unsigned int _num_threads_hint = {}; }; -- cgit v1.2.1