diff options
author | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2022-01-27 17:39:06 +0100 |
---|---|---|
committer | Kristofer Jonsson <kristofer.jonsson@arm.com> | 2022-01-31 16:58:33 +0100 |
commit | 5410db1be8c1b8f84c5062e646156e8bca42a4c4 (patch) | |
tree | f782ab4ee6bbc9e15643733ad937a564def8cbe7 /applications/message_handler/message_handler.hpp | |
parent | 6e9fdc0565ff07fce1fb5efc18bb7d5dbe3da489 (diff) | |
download | ethos-u-core-platform-5410db1be8c1b8f84c5062e646156e8bca42a4c4.tar.gz |
PMU multi NPU support
Update sample applications to use an external context for the PMU
configuration. The external context stored in the InferenceJob will
be set as TFLu external context and will be returned in the
ethosu_inference_begin() and ethosu_inference_end() callbacks.
Change-Id: Ief1f0943e322c2b50e8b964017af59161f67de6b
Diffstat (limited to 'applications/message_handler/message_handler.hpp')
-rw-r--r-- | applications/message_handler/message_handler.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/applications/message_handler/message_handler.hpp b/applications/message_handler/message_handler.hpp index 5a195f0..36768ee 100644 --- a/applications/message_handler/message_handler.hpp +++ b/applications/message_handler/message_handler.hpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2021 Arm Limited. All rights reserved. + * Copyright (c) 2020-2022 Arm Limited. All rights reserved. * * SPDX-License-Identifier: Apache-2.0 * @@ -24,6 +24,7 @@ #include "semphr.h" #include "message_queue.hpp" +#include <ethosu_driver.h> #include <inference_process.hpp> #include <mailbox.hpp> @@ -62,9 +63,14 @@ public: private: void runInference(EthosU::ethosu_core_inference_req &req, EthosU::ethosu_core_inference_rsp &rsp); + friend void ::ethosu_inference_begin(struct ethosu_driver *drv, void *userArg); + friend void ::ethosu_inference_end(struct ethosu_driver *drv, void *userArg); + QueueHandle_t inferenceQueue; QueueHandle_t outputQueue; InferenceProcess::InferenceProcess inference; + EthosU::ethosu_core_inference_req *currentReq; + EthosU::ethosu_core_inference_rsp *currentRsp; }; struct OutputMessage { |