aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/CPP/CPPScheduler.h
diff options
context:
space:
mode:
authorMoritz Pflanzer <moritz.pflanzer@arm.com>2017-06-28 10:15:48 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-09-17 14:16:42 +0100
commitd929b9c49a13eb9c05bb4fab608459669eeeeb9e (patch)
treeed0101971fe0d8c112f715d47a53aae306031d93 /arm_compute/runtime/CPP/CPPScheduler.h
parent725788e33c163f95a6fdc054b6b78296ed7dfe50 (diff)
downloadComputeLibrary-d929b9c49a13eb9c05bb4fab608459669eeeeb9e.tar.gz
COMPMID-417: Enable CPU target selection
Change-Id: I8d1b368c654b738117efb32cfacd5fda10c23203 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79046 Reviewed-by: Anthony Barbier <anthony.barbier@arm.com> Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'arm_compute/runtime/CPP/CPPScheduler.h')
-rw-r--r--arm_compute/runtime/CPP/CPPScheduler.h15
1 files changed, 15 insertions, 0 deletions
diff --git a/arm_compute/runtime/CPP/CPPScheduler.h b/arm_compute/runtime/CPP/CPPScheduler.h
index 7a37e5ec21..341537a390 100644
--- a/arm_compute/runtime/CPP/CPPScheduler.h
+++ b/arm_compute/runtime/CPP/CPPScheduler.h
@@ -24,6 +24,7 @@
#ifndef __ARM_COMPUTE_CPPSCHEDULER_H__
#define __ARM_COMPUTE_CPPSCHEDULER_H__
+#include "arm_compute/core/CPP/CPPTypes.h"
#include "arm_compute/runtime/IScheduler.h"
#include <memory>
@@ -46,6 +47,19 @@ public:
* @return Number of threads available in CPPScheduler.
*/
unsigned int num_threads() const override;
+
+ /** Sets the target CPU architecture.
+ *
+ * @param[in] target Target CPU.
+ */
+ void set_target(CPUTarget target);
+
+ /** Return the current CPU target.
+ *
+ * @return Target CPU.
+ */
+ CPUTarget target() const;
+
/** Access the scheduler singleton
*
* @return The scheduler
@@ -68,6 +82,7 @@ private:
unsigned int _num_threads;
std::unique_ptr<Thread[], void (*)(Thread *)> _threads;
+ CPUTarget _target;
};
}
#endif /* __ARM_COMPUTE_CPPSCHEDULER_H__ */