aboutsummaryrefslogtreecommitdiff
path: root/applications/inference_process/include/inference_process.hpp
diff options
context:
space:
mode:
authorAnton Moberg <anton.moberg@arm.com>2021-02-10 08:49:28 +0100
committerAnton Moberg <anton.moberg@arm.com>2021-02-12 14:37:30 +0100
commit66ed182cd7520537e73ec37f17f8bf549d8297a2 (patch)
tree7ce2ce2f2c10f24d3b6fd5b4cb3e3bd9024ad2a3 /applications/inference_process/include/inference_process.hpp
parent83e49967d1c4eeff21025ea0bd449c938c91c5f8 (diff)
downloadethos-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.hpp5
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();