diff options
author | Mike Kelly <mike.kelly@arm.com> | 2021-06-27 22:39:21 +0100 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2021-06-30 10:38:30 +0000 |
commit | 0d677db72eb7945e304fc49cedf744f0c34ed330 (patch) | |
tree | ee85b7f50e4ad9d5e568b46df46777513a2197cd /src/armnn/test | |
parent | 6f24b1aac9cbf5dfb3918cb59a7e903ddcec420e (diff) | |
download | armnn-0d677db72eb7945e304fc49cedf744f0c34ed330.tar.gz |
IVGCVSW-6114 Create multiple LoadedNetworks from one OptimizedNetwork
* Added IOptimizedNetwork constructor that takes another
IOptimizedNetwork and a ModelOptions.
* Changed PreCompiledLayer to use shared_ptr rather than unique_ptr
to store the PreCompiledObject (no interface changes).
* Added unit tests to ensure that PreCompiledLayer::Clone() clones
the pointer to the PreCompiledObject correctly.
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I3ef56055e0d189ffce9e651882d34da16c70a240
Diffstat (limited to 'src/armnn/test')
-rw-r--r-- | src/armnn/test/CloneTests.cpp | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/src/armnn/test/CloneTests.cpp b/src/armnn/test/CloneTests.cpp new file mode 100644 index 0000000000..2ee2cdad0e --- /dev/null +++ b/src/armnn/test/CloneTests.cpp @@ -0,0 +1,103 @@ +// +// Copyright © 2021 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include <Graph.hpp> +#include <Layer.hpp> + +#include <armnn/TypesUtils.hpp> +#include <armnn/Exceptions.hpp> +#include <armnn/Optional.hpp> +#include <armnn/backends/IBackendInternal.hpp> +#include <armnn/utility/PolymorphicDowncast.hpp> + +#include <backendsCommon/TensorHandle.hpp> +#include <backendsCommon/WorkloadFactoryBase.hpp> + +#include <doctest/doctest.h> + +namespace { + +const armnn::BackendId& GetCloneIdStatic() +{ + static const armnn::BackendId s_Id{"Tests"}; + return s_Id; +} + +class TestWorkloadFactory : public armnn::WorkloadFactoryBase +{ +public: + + TestWorkloadFactory() + : m_Ptr(nullptr) + {} + + const armnn::BackendId& GetBackendId() const override + { + return GetCloneIdStatic(); + } + + std::unique_ptr<armnn::IWorkload> CreatePreCompiled(const armnn::PreCompiledQueueDescriptor& descriptor, + const armnn::WorkloadInfo&) const override + { + if (m_Ptr) + { + CHECK(descriptor.m_PreCompiledObject == m_Ptr); + } + else + { + m_Ptr = descriptor.m_PreCompiledObject; + } + return nullptr; + } + + mutable void* m_Ptr; +}; + +TEST_SUITE("CloneTests") +{ + +TEST_CASE ("PreCompiledLayerClonePreservesObject") +{ + armnn::Graph graph1; + armnn::Graph graph2; + + armnn::PreCompiledDescriptor descriptor(0u, 0u); + + armnn::Layer* const preCompiledLayer = graph1.AddLayer<armnn::PreCompiledLayer>(descriptor, "preCompiled"); + armnn::PreCompiledLayer* layer = armnn::PolymorphicDowncast<armnn::PreCompiledLayer*>(preCompiledLayer); + + armnn::PreCompiledObjectPtr payloadObject; + TestWorkloadFactory factory; + + layer->SetPreCompiledObject(std::move(payloadObject)); + layer->CreateWorkload(factory); + + armnn::PreCompiledLayer* clone = layer->Clone(graph2); + CHECK(std::strcmp(clone->GetName(), "preCompiled") == 0); + clone->CreateWorkload(factory); +} + +TEST_CASE ("PreCompiledLayerCloneNoObject") +{ + armnn::Graph graph1; + + armnn::Graph graph2; + + armnn::PreCompiledDescriptor descriptor(0u, 0u); + + armnn::Layer* const preCompiledLayer = graph1.AddLayer<armnn::PreCompiledLayer>(descriptor, "preCompiled"); + armnn::PreCompiledLayer* layer = armnn::PolymorphicDowncast<armnn::PreCompiledLayer*>(preCompiledLayer); + + TestWorkloadFactory factory; + layer->CreateWorkload(factory); + + armnn::PreCompiledLayer* clone = layer->Clone(graph2); + CHECK(std::strcmp(clone->GetName(), "preCompiled") == 0); + clone->CreateWorkload(factory); +} + +} + +} // end anonymous namespace |