From c42a987aa53d0fd842c34dee90abef5a9ff15fa4 Mon Sep 17 00:00:00 2001
From: Colm Donelan <colm.donelan@arm.com>
Date: Wed, 2 Feb 2022 16:35:09 +0000
Subject: IVGCVSW-6635 Move MemCopyTestImpl from acl to armnnTestUtils.

* Move MemCopyTestImpl.hpp from src/backends/aclCommon/test/ to
  include/armnnTestutils.
* Refactor MemCopyTests in aclCommon, cl and Neon.
* Introduce RefMemCopyTests to exercise this utility in x86 builds.

Signed-off-by: Colm Donelan <colm.donelan@arm.com>
Change-Id: I8824f013d3656658ed0a2904bb79384e3af68641
---
 .../armnnTestUtils/MockWorkloadFactoryHelper.hpp   | 41 ++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 include/armnnTestUtils/MockWorkloadFactoryHelper.hpp

(limited to 'include/armnnTestUtils/MockWorkloadFactoryHelper.hpp')

diff --git a/include/armnnTestUtils/MockWorkloadFactoryHelper.hpp b/include/armnnTestUtils/MockWorkloadFactoryHelper.hpp
new file mode 100644
index 0000000000..e9f2b2f6cf
--- /dev/null
+++ b/include/armnnTestUtils/MockWorkloadFactoryHelper.hpp
@@ -0,0 +1,41 @@
+//
+// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "MockBackend.hpp"
+#include "MockTensorHandleFactory.hpp"
+#include <backendsCommon/test/WorkloadFactoryHelper.hpp>
+
+namespace
+{
+
+template <>
+struct WorkloadFactoryHelper<armnn::MockWorkloadFactory>
+{
+    static armnn::IBackendInternal::IMemoryManagerSharedPtr GetMemoryManager()
+    {
+        armnn::MockBackend backend;
+        return backend.CreateMemoryManager();
+    }
+
+    static armnn::MockWorkloadFactory
+        GetFactory(const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager = nullptr)
+    {
+        IgnoreUnused(memoryManager);
+        return armnn::MockWorkloadFactory();
+    }
+
+    static armnn::MockTensorHandleFactory
+        GetTensorHandleFactory(const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager = nullptr)
+    {
+
+        return armnn::MockTensorHandleFactory(std::static_pointer_cast<armnn::MockMemoryManager>(memoryManager));
+    }
+};
+
+using MockWorkloadFactoryHelper = WorkloadFactoryHelper<armnn::MockWorkloadFactory>;
+
+}    // anonymous namespace
-- 
cgit v1.2.1