From 60578950322491e44b4203fe085c3230ead19c7a Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Wed, 31 Oct 2018 11:04:01 +0000 Subject: IVGCVSW-1709: Clean up memory manager bodge * Added quick workaround for the memory management errors in the CL/Neon unit tests Change-Id: I56250db462cdbdc1acc0a5824807d288fb0c1d11 --- src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp') diff --git a/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp b/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp index 125b7e62b1..298901e640 100644 --- a/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp +++ b/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp @@ -41,8 +41,9 @@ LayerTestResult SimpleFullyConnectedTestImpl( outputHandle->Allocate(); CopyDataToITensorHandle(inputHandle.get(), &input[0][0][0][0]); - workloadFactory.Finalize(); + workloadFactory.Acquire(); workload->Execute(); + workloadFactory.Release(); CopyDataFromITensorHandle(&result.output[0][0], outputHandle.get()); -- cgit v1.2.1