IVGCVSW-6635 Expose a new MockWorkloadFactory and MockMemManager (Part 1)

* Create a MockBackend in armnnTestUtils.
* Create corresponding WorkloadFactory, TensorHandle, TensorHandleFactory
  MemoryManager and WorkloadFactoryHelper classes.

Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: I884731b109bc623a7272e5ad333ff754f8c13ae1

1 files changed, 55 insertions, 0 deletions

diff --git a/src/armnnTestUtils/MockBackend.cpp b/src/armnnTestUtils/MockBackend.cpp
new file mode 100644
index 0000000000..40f97cd28e
--- /dev/null
+++ b/src/armnnTestUtils/MockBackend.cpp
@@ -0,0 +1,55 @@
+//
+// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include <armnn/backends/MemCopyWorkload.hpp>
+#include <armnnTestUtils/MockBackend.hpp>
+#include <armnnTestUtils/MockTensorHandle.hpp>
+
+namespace armnn
+{
+
+constexpr const char* MockBackendId()
+{
+    return "CpuMock";
+}
+
+namespace
+{
+static const BackendId s_Id{ MockBackendId() };
+}
+
+MockWorkloadFactory::MockWorkloadFactory(const std::shared_ptr<MockMemoryManager>& memoryManager)
+    : m_MemoryManager(memoryManager)
+{}
+
+MockWorkloadFactory::MockWorkloadFactory()
+    : m_MemoryManager(new MockMemoryManager())
+{}
+
+const BackendId& MockWorkloadFactory::GetBackendId() const
+{
+    return s_Id;
+}
+
+std::unique_ptr<IWorkload> MockWorkloadFactory::CreateWorkload(LayerType type,
+                                                               const QueueDescriptor& descriptor,
+                                                               const WorkloadInfo& info) const
+{
+    switch (type)
+    {
+        case LayerType::MemCopy: {
+            auto memCopyQueueDescriptor = PolymorphicDowncast<const MemCopyQueueDescriptor*>(&descriptor);
+            if (descriptor.m_Inputs.empty())
+            {
+                throw InvalidArgumentException("MockWorkloadFactory: CreateMemCopy() expected an input tensor.");
+            }
+            return std::make_unique<CopyMemGenericWorkload>(*memCopyQueueDescriptor, info);
+        }
+        default:
+            return nullptr;
+    }
+}
+
+}    // namespace armnn
\ No newline at end of file