diff options
Diffstat (limited to 'arm_compute/runtime/CPP/CPPScheduler.h')
| -rw-r--r-- | arm_compute/runtime/CPP/CPPScheduler.h | 15 |
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__ */ |