diff options
author | Anton Moberg <anton.moberg@arm.com> | 2021-02-10 08:49:28 +0100 |
---|---|---|
committer | Anton Moberg <anton.moberg@arm.com> | 2021-02-12 14:37:30 +0100 |
commit | 66ed182cd7520537e73ec37f17f8bf549d8297a2 (patch) | |
tree | 7ce2ce2f2c10f24d3b6fd5b4cb3e3bd9024ad2a3 /applications/inference_process/include/inference_process.hpp | |
parent | 83e49967d1c4eeff21025ea0bd449c938c91c5f8 (diff) | |
download | ethos-u-core-software-66ed182cd7520537e73ec37f17f8bf549d8297a2.tar.gz |
core_software - Move TensorArena21.02-rc2
Moved TensorArena in inference_process.ccp to application level.
InferenceProcess class now takes TensorArena pointer and TensorArenaSize as parameters. Needs to be set by application before runJob() is called.
Change-Id: I530b96039868305fa903ae7f93419d9d00f9c16f
Diffstat (limited to 'applications/inference_process/include/inference_process.hpp')
-rw-r--r-- | applications/inference_process/include/inference_process.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/applications/inference_process/include/inference_process.hpp b/applications/inference_process/include/inference_process.hpp index 67b30c5..880e28d 100644 --- a/applications/inference_process/include/inference_process.hpp +++ b/applications/inference_process/include/inference_process.hpp @@ -63,7 +63,8 @@ struct InferenceJob { class InferenceProcess { public: - InferenceProcess(); + InferenceProcess(uint8_t *_tensorArena, size_t _tensorArenaSize) : + lock(0), tensorArena(_tensorArena), tensorArenaSize(_tensorArenaSize) {} bool push(const InferenceJob &job); bool runJob(InferenceJob &job); @@ -71,6 +72,8 @@ public: private: volatile uint32_t lock; + uint8_t *tensorArena; + const size_t tensorArenaSize; std::queue<InferenceJob> inferenceJobQueue; void getLock(); |