diff options
Diffstat (limited to 'src/runtime/CPP')
-rw-r--r-- | src/runtime/CPP/CPPScheduler.cpp | 23 |
1 files changed, 17 insertions, 6 deletions
diff --git a/src/runtime/CPP/CPPScheduler.cpp b/src/runtime/CPP/CPPScheduler.cpp index 886933074d..2a321a1101 100644 --- a/src/runtime/CPP/CPPScheduler.cpp +++ b/src/runtime/CPP/CPPScheduler.cpp @@ -161,14 +161,10 @@ CPPScheduler &CPPScheduler::get() return scheduler; } -unsigned int CPPScheduler::num_threads() const -{ - return _num_threads; -} - CPPScheduler::CPPScheduler() : _num_threads(std::thread::hardware_concurrency()), - _threads(std::unique_ptr<Thread[], void(*)(Thread *)>(new Thread[std::thread::hardware_concurrency() - 1], delete_threads)) + _threads(std::unique_ptr<Thread[], void(*)(Thread *)>(new Thread[std::thread::hardware_concurrency() - 1], delete_threads)), + _target(CPUTarget::INTRINSICS) { } @@ -178,6 +174,21 @@ void CPPScheduler::set_num_threads(unsigned int num_threads) _num_threads = num_threads == 0 ? num_cores : num_threads; } +unsigned int CPPScheduler::num_threads() const +{ + return _num_threads; +} + +void CPPScheduler::set_target(CPUTarget target) +{ + _target = target; +} + +CPUTarget CPPScheduler::target() const +{ + return _target; +} + void CPPScheduler::schedule(ICPPKernel *kernel, unsigned int split_dimension) { ARM_COMPUTE_ERROR_ON_MSG(!kernel, "The child class didn't set the kernel"); |