diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-06-09 08:45:18 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2022-06-10 11:47:47 +0000 |
commit | f98421efb1ab438dc02b0cf7abc670f8a29750f4 (patch) | |
tree | d9507c86029ce47b2488e2aa4d80d31b8bf88a3f | |
parent | 4788c384f738cc8eb659fb3838132d7c59d6aab7 (diff) | |
download | armnn-f98421efb1ab438dc02b0cf7abc670f8a29750f4.tar.gz |
IVGCVSW-6872 SLTS Failures and GpuAcc takes too long to complete
* Fix memory issue by calling UnloadNetwork() in destructor of ArmnnPreparedModel.
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: I255a50f2bf0bfcc8eedf1ad34cea9249a046bce8
-rw-r--r-- | shim/sl/canonical/ArmnnPreparedModel.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/shim/sl/canonical/ArmnnPreparedModel.cpp b/shim/sl/canonical/ArmnnPreparedModel.cpp index 910a0fce46..7b900a9418 100644 --- a/shim/sl/canonical/ArmnnPreparedModel.cpp +++ b/shim/sl/canonical/ArmnnPreparedModel.cpp @@ -614,6 +614,8 @@ ArmnnPreparedModel::~ArmnnPreparedModel() profiler.get()); } } + // Unload the network associated with this model + m_Runtime->UnloadNetwork(m_NetworkId); } bool ArmnnPreparedModel::ExecuteWithDummyInputs(unsigned int numInputs, unsigned int numOutputs) const |