From 8c999dfeeca7b02a6ea1d0cdcd8c34472f6c9cce Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Wed, 4 Aug 2021 17:27:40 +0100 Subject: Fixed bugs in PreCompiledLayer * Fixed PreCompiledObject type (was const void** instead of void*) * Fixed bug where a new shared_ptr was being created instead of allowing std::move to convert the unique_ptr into a shared_ptr. * Improved tests to ensure that the original pointer is maintained. Signed-off-by: Mike Kelly Change-Id: I3f50d51775ee0180c894c2843ed7bc990a771dea --- src/armnn/layers/PreCompiledLayer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/armnn/layers/PreCompiledLayer.cpp') diff --git a/src/armnn/layers/PreCompiledLayer.cpp b/src/armnn/layers/PreCompiledLayer.cpp index 0cc9c5a75a..67c1db4011 100644 --- a/src/armnn/layers/PreCompiledLayer.cpp +++ b/src/armnn/layers/PreCompiledLayer.cpp @@ -46,7 +46,7 @@ void PreCompiledLayer::ValidateTensorShapesFromInputs() void PreCompiledLayer::SetPreCompiledObject(PreCompiledObjectPtr preCompiledObject) { - m_PreCompiledObject = std::make_shared(preCompiledObject.release()); + m_PreCompiledObject = std::move(preCompiledObject); } void PreCompiledLayer::Accept(ILayerVisitor& visitor) const -- cgit v1.2.1