diff options
Diffstat (limited to 'arm_compute/runtime/Scheduler.h')
-rw-r--r-- | arm_compute/runtime/Scheduler.h | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/arm_compute/runtime/Scheduler.h b/arm_compute/runtime/Scheduler.h index 7e10461b5a..89263fd176 100644 --- a/arm_compute/runtime/Scheduler.h +++ b/arm_compute/runtime/Scheduler.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -25,6 +25,8 @@ #define __ARM_COMPUTE_SCHEDULER_H__ #include "arm_compute/runtime/IScheduler.h" + +#include <map> #include <memory> namespace arm_compute @@ -74,7 +76,9 @@ public: private: static Type _scheduler_type; static std::shared_ptr<IScheduler> _custom_scheduler; + static std::map<Type, std::unique_ptr<IScheduler>> _schedulers; + Scheduler(); }; -} +} // namespace arm_compute #endif /* __ARM_COMPUTE_SCHEDULER_H__ */ |