diff options
author | Jim Flynn <jim.flynn@arm.com> | 2020-10-06 10:14:50 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-10-07 14:02:38 +0000 |
commit | 68db06f466896d063822904915b24190cab92284 (patch) | |
tree | 8636bf70ffed2a4ab40083f8dfe1d4ebfcb817c3 /src/backends/backendsCommon | |
parent | 84dc8430432c0a65c2fdc8946e06d307d18df28d (diff) | |
download | armnn-68db06f466896d063822904915b24190cab92284.tar.gz |
IVGCVSW-5362 Add Map layer and Map workload
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Id2227c58809b84c7a7af61f7c0d88ad7d45ce558
Diffstat (limited to 'src/backends/backendsCommon')
-rw-r--r-- | src/backends/backendsCommon/CMakeLists.txt | 2 | ||||
-rw-r--r-- | src/backends/backendsCommon/MapWorkload.cpp | 22 | ||||
-rw-r--r-- | src/backends/backendsCommon/MapWorkload.hpp | 19 | ||||
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 18 | ||||
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.hpp | 5 | ||||
-rw-r--r-- | src/backends/backendsCommon/common.mk | 1 | ||||
-rw-r--r-- | src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp | 47 |
7 files changed, 114 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/CMakeLists.txt b/src/backends/backendsCommon/CMakeLists.txt index cec56f15dd..28b3088e48 100644 --- a/src/backends/backendsCommon/CMakeLists.txt +++ b/src/backends/backendsCommon/CMakeLists.txt @@ -22,6 +22,8 @@ list(APPEND armnnBackendsCommon_sources LayerSupportBase.hpp LayerSupportRules.hpp MakeWorkloadHelper.hpp + MapWorkload.cpp + MapWorkload.hpp MemCopyWorkload.cpp MemCopyWorkload.hpp MemImportWorkload.cpp diff --git a/src/backends/backendsCommon/MapWorkload.cpp b/src/backends/backendsCommon/MapWorkload.cpp new file mode 100644 index 0000000000..99dd43e61d --- /dev/null +++ b/src/backends/backendsCommon/MapWorkload.cpp @@ -0,0 +1,22 @@ +// +// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include <backendsCommon/MapWorkload.hpp> + +namespace armnn +{ + +MapWorkload::MapWorkload(const MapQueueDescriptor& descriptor, + const WorkloadInfo& info) + : BaseWorkload<MapQueueDescriptor>(descriptor, info) +{ +} + +void MapWorkload::Execute() const +{ + m_Data.m_Inputs[0]->Map(); +} + +} //namespace armnn diff --git a/src/backends/backendsCommon/MapWorkload.hpp b/src/backends/backendsCommon/MapWorkload.hpp new file mode 100644 index 0000000000..dbbf36a9ee --- /dev/null +++ b/src/backends/backendsCommon/MapWorkload.hpp @@ -0,0 +1,19 @@ +// +// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// +#pragma once + +#include "Workload.hpp" + +namespace armnn +{ + +class MapWorkload : public BaseWorkload<MapQueueDescriptor> +{ +public: + MapWorkload(const MapQueueDescriptor& descriptor, const WorkloadInfo& info); + void Execute() const override; +}; + +} //namespace armnn diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp index e9023b62e1..134495991e 100644 --- a/src/backends/backendsCommon/WorkloadData.cpp +++ b/src/backends/backendsCommon/WorkloadData.cpp @@ -464,6 +464,24 @@ void QueueDescriptor::ValidateInputsOutputs(const std::string& descName, } //--------------------------------------------------------------- +void MapQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const +{ + const std::string descriptorName{"MapQueueDescriptor"}; + + ValidateNumInputs(workloadInfo, descriptorName, 1); + ValidateNumOutputs(workloadInfo, descriptorName , 0); + + for (unsigned int i = 0; i < m_Inputs.size(); ++i) + { + if (!m_Inputs[i]) + { + throw InvalidArgumentException( + fmt::format("{}: Invalid NULL input {}.", descriptorName, static_cast<int>(i))); + } + } +} + +//--------------------------------------------------------------- void MemCopyQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const { const std::string descriptorName{"MemCopyQueueDescriptor"}; diff --git a/src/backends/backendsCommon/WorkloadData.hpp b/src/backends/backendsCommon/WorkloadData.hpp index 1f54f9aaed..be0a67e753 100644 --- a/src/backends/backendsCommon/WorkloadData.hpp +++ b/src/backends/backendsCommon/WorkloadData.hpp @@ -55,6 +55,11 @@ protected: QueueDescriptorWithParameters& operator=(QueueDescriptorWithParameters const&) = default; }; +struct MapQueueDescriptor : QueueDescriptor +{ + void Validate(const WorkloadInfo& workloadInfo) const; +}; + struct MemCopyQueueDescriptor : QueueDescriptor { void Validate(const WorkloadInfo& workloadInfo) const; diff --git a/src/backends/backendsCommon/common.mk b/src/backends/backendsCommon/common.mk index 7829abbbb9..ceec2ac6e1 100644 --- a/src/backends/backendsCommon/common.mk +++ b/src/backends/backendsCommon/common.mk @@ -14,6 +14,7 @@ COMMON_SOURCES := \ IBackendInternal.cpp \ ITensorHandleFactory.cpp \ LayerSupportBase.cpp \ + MapWorkload.cpp \ MemCopyWorkload.cpp \ MemImportWorkload.cpp \ MemSyncWorkload.cpp \ diff --git a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp index ed4b6ff09a..1078c2a8d7 100644 --- a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp +++ b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp @@ -6,6 +6,7 @@ #include <Graph.hpp> +#include <backendsCommon/MapWorkload.hpp> #include <backendsCommon/WorkloadFactory.hpp> #include <armnn/utility/IgnoreUnused.hpp> @@ -165,6 +166,22 @@ struct DummyLayer<armnn::ConcatLayer> }; template<> +struct DummyLayer<armnn::MapLayer, void> +{ + DummyLayer() + { + m_Layer = dummyGraph.AddLayer<armnn::MapLayer>(""); + } + + ~DummyLayer() + { + dummyGraph.EraseLayer(m_Layer); + } + + armnn::MapLayer* m_Layer; +}; + +template<> struct DummyLayer<armnn::OutputLayer, armnn::LayerBindingId> { DummyLayer() @@ -467,6 +484,27 @@ struct LayerTypePolicy<armnn::LayerType::name, DataType> \ } \ }; +#define DECLARE_LAYER_POLICY_MAP_PARAM(name, descType) \ +template<armnn::DataType DataType> \ +struct LayerTypePolicy<armnn::LayerType::name, DataType> \ +{ \ + using Type = armnn::name##Layer; \ + using Desc = descType; \ + using QueueDesc = armnn::name##QueueDescriptor; \ + using Workload = armnn::name##Workload; \ + constexpr static const char* NameStr = #name; \ + constexpr static const bool IsException = false; \ + \ + static std::unique_ptr<armnn::IWorkload> MakeDummyWorkload(armnn::IWorkloadFactory* factory, \ + unsigned int nIn, unsigned int nOut) \ + { \ + IgnoreUnused(factory); \ + QueueDesc desc; \ + armnn::WorkloadInfo info = MakeDummyWorkloadInfo<DataType>(nIn, nOut); \ + return std::make_unique<armnn::name##Workload>(desc, info); \ + } \ +}; + // Define a layer policy specialization for use with the IsLayerSupported tests. // Use this version for layers whose constructor takes 1 parameter(name). #define DECLARE_LAYER_POLICY_1_PARAM(name) DECLARE_LAYER_POLICY_CUSTOM_PARAM(name, void) @@ -563,6 +601,8 @@ DECLARE_LAYER_POLICY_2_PARAM(LogSoftmax) DECLARE_LAYER_POLICY_2_PARAM(Lstm) +DECLARE_LAYER_POLICY_MAP_PARAM(Map, void) + DECLARE_LAYER_POLICY_1_PARAM(Maximum) DECLARE_LAYER_POLICY_2_PARAM(Mean) @@ -751,6 +791,13 @@ bool IsLayerSupportedTest(FactoryType *factory, Tag<Type>) } } +template<typename FactoryType, armnn::DataType DataType, armnn::LayerType Type> +bool IsLayerSupportedTest(FactoryType *factory, Tag<armnn::LayerType::Map>) +{ + IgnoreUnused(factory); + return true; +} + // Helper function to compute the next type in the LayerType enum. constexpr armnn::LayerType NextType(armnn::LayerType type) { |