aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/CPP/CPPScheduler.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/CPP/CPPScheduler.cpp')
-rw-r--r--src/runtime/CPP/CPPScheduler.cpp23
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");