From f98421efb1ab438dc02b0cf7abc670f8a29750f4 Mon Sep 17 00:00:00 2001 From: Cathal Corbett Date: Thu, 9 Jun 2022 08:45:18 +0100 Subject: 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 Change-Id: I255a50f2bf0bfcc8eedf1ad34cea9249a046bce8 --- shim/sl/canonical/ArmnnPreparedModel.cpp | 2 ++ 1 file changed, 2 insertions(+) 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 -- cgit v1.2.1