From 66ed182cd7520537e73ec37f17f8bf549d8297a2 Mon Sep 17 00:00:00 2001
From: Anton Moberg <anton.moberg@arm.com>
Date: Wed, 10 Feb 2021 08:49:28 +0100
Subject: core_software - Move TensorArena

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
---
 applications/inference_process/include/inference_process.hpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'applications/inference_process/include/inference_process.hpp')

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();
-- 
cgit v1.2.1