From 17948b5eb666f0ca17b40ccaf6712361bbfe6a44 Mon Sep 17 00:00:00 2001
From: Colm Donelan <colm.donelan@arm.com>
Date: Tue, 1 Feb 2022 23:37:04 +0000
Subject: 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
---
 src/armnnTestUtils/MockMemoryManager.cpp | 93 ++++++++++++++++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 src/armnnTestUtils/MockMemoryManager.cpp

(limited to 'src/armnnTestUtils/MockMemoryManager.cpp')

diff --git a/src/armnnTestUtils/MockMemoryManager.cpp b/src/armnnTestUtils/MockMemoryManager.cpp
new file mode 100644
index 0000000000..3ff6ec85d9
--- /dev/null
+++ b/src/armnnTestUtils/MockMemoryManager.cpp
@@ -0,0 +1,93 @@
+//
+// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "armnnTestUtils/MockMemoryManager.hpp"
+
+namespace armnn
+{
+
+MockMemoryManager::MockMemoryManager()
+{}
+
+MockMemoryManager::~MockMemoryManager()
+{}
+
+MockMemoryManager::Pool* MockMemoryManager::Manage(unsigned int numBytes)
+{
+    if (!m_FreePools.empty())
+    {
+        Pool* res = m_FreePools.back();
+        m_FreePools.pop_back();
+        res->Reserve(numBytes);
+        return res;
+    }
+    else
+    {
+        m_Pools.push_front(Pool(numBytes));
+        return &m_Pools.front();
+    }
+}
+
+void MockMemoryManager::Allocate(MockMemoryManager::Pool* pool)
+{
+    m_FreePools.push_back(pool);
+}
+
+void* MockMemoryManager::GetPointer(MockMemoryManager::Pool* pool)
+{
+    return pool->GetPointer();
+}
+
+void MockMemoryManager::Acquire()
+{
+    for (Pool& pool : m_Pools)
+    {
+        pool.Acquire();
+    }
+}
+
+void MockMemoryManager::Release()
+{
+    for (Pool& pool : m_Pools)
+    {
+        pool.Release();
+    }
+}
+
+MockMemoryManager::Pool::Pool(unsigned int numBytes)
+    : m_Size(numBytes)
+    , m_Pointer(nullptr)
+{}
+
+MockMemoryManager::Pool::~Pool()
+{
+    if (m_Pointer)
+    {
+        Release();
+    }
+}
+
+void* MockMemoryManager::Pool::GetPointer()
+{
+    return m_Pointer;
+}
+
+void MockMemoryManager::Pool::Reserve(unsigned int numBytes)
+{
+    m_Size = std::max(m_Size, numBytes);
+}
+
+void MockMemoryManager::Pool::Acquire()
+{
+    m_Pointer = ::operator new(size_t(m_Size));
+}
+
+void MockMemoryManager::Pool::Release()
+{
+    ::operator delete(m_Pointer);
+    m_Pointer = nullptr;
+}
+
+}    // namespace armnn
-- 
cgit v1.2.1