aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/main.cpp
diff options
context:
space:
mode:
authorJerry Ge <jerry.ge@arm.com>2023-07-17 21:33:17 +0000
committerJerry Ge <jerry.ge@arm.com>2023-07-27 17:24:26 +0000
commite5cabbf7528849aac35b498ce0711a144c1a08d5 (patch)
treec4d4756a5961cd095a4f0faa677dbe6b47522f82 /reference_model/src/main.cpp
parent97b0027ca018ad9b5f91f41b413e843afb15c6d7 (diff)
downloadreference_model-e5cabbf7528849aac35b498ce0711a144c1a08d5.tar.gz
Enable lazy tensor allocation
- The previous ref_model was allocating the memory for all tensors in the graph upfront which is unnecessary and wasteful. - This patch changes to only allocate initial input tensors on the entry point using the allocateInputTensor() function. - The output tensors are ensured to have been allocated before executing a node. The output tenosrs are the inputs for the next node. - When a node's evaluation is finished, its input tensors will be freed if they will no longer be used by anyone else. Signed-off-by: Jerry Ge <jerry.ge@arm.com> Change-Id: Ibb3e8c9e6344f6cd9eb20532a03b2097b93247f9
Diffstat (limited to 'reference_model/src/main.cpp')
-rw-r--r--reference_model/src/main.cpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/reference_model/src/main.cpp b/reference_model/src/main.cpp
index 0c86cbd..070eb33 100644
--- a/reference_model/src/main.cpp
+++ b/reference_model/src/main.cpp
@@ -109,9 +109,9 @@ int main(int argc, char** argv)
goto done;
}
- if (main_gt.allocateTensor())
+ if (main_gt.allocateInputTensors())
{
- WARNING("Failed to allocate tensor. Evaluation aborted.");
+ WARNING("Failed to allocate input tensors. Evaluation aborted.");
goto done;
}