From d929b9c49a13eb9c05bb4fab608459669eeeeb9e Mon Sep 17 00:00:00 2001
From: Moritz Pflanzer <moritz.pflanzer@arm.com>
Date: Wed, 28 Jun 2017 10:15:48 +0100
Subject: 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>
---
 arm_compute/runtime/CPP/CPPScheduler.h | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'arm_compute/runtime/CPP/CPPScheduler.h')

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__ */
-- 
cgit v1.2.1