IVGCVSW-3307 Add RefMemoryManager

Simple pool memory manager for use in the reference backend,
in order to make the backend usable for testing large networks.

Change-Id: I5694da29052c60f95b57da595c64cc114d75b8ba
Signed-off-by: Matthew Bentham <Matthew.Bentham@arm.com>

1 files changed, 100 insertions, 0 deletions

diff --git a/src/backends/reference/RefMemoryManager.cpp b/src/backends/reference/RefMemoryManager.cpp
new file mode 100644
index 0000000000..0f4a289807
--- /dev/null
+++ b/src/backends/reference/RefMemoryManager.cpp
@@ -0,0 +1,100 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#include "RefMemoryManager.hpp"
+
+#include <boost/assert.hpp>
+
+namespace armnn
+{
+
+RefMemoryManager::RefMemoryManager()
+{}
+
+RefMemoryManager::~RefMemoryManager()
+{}
+
+RefMemoryManager::Pool* RefMemoryManager::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 RefMemoryManager::Allocate(RefMemoryManager::Pool* pool)
+{
+    BOOST_ASSERT(pool);
+    m_FreePools.push_back(pool);
+}
+
+void* RefMemoryManager::GetPointer(RefMemoryManager::Pool* pool)
+{
+    return pool->GetPointer();
+}
+
+void RefMemoryManager::Acquire()
+{
+    for (Pool &pool: m_Pools)
+    {
+         pool.Acquire();
+    }
+}
+
+void RefMemoryManager::Release()
+{
+    for (Pool &pool: m_Pools)
+    {
+         pool.Release();
+    }
+}
+
+RefMemoryManager::Pool::Pool(unsigned int numBytes)
+    : m_Size(numBytes),
+      m_Pointer(nullptr)
+{}
+
+RefMemoryManager::Pool::~Pool()
+{
+    if (m_Pointer)
+    {
+        Release();
+    }
+}
+
+void* RefMemoryManager::Pool::GetPointer()
+{
+    BOOST_ASSERT_MSG(m_Pointer, "RefMemoryManager::Pool::GetPointer() called when memory not acquired"); 
+    return m_Pointer;
+}
+
+void RefMemoryManager::Pool::Reserve(unsigned int numBytes)
+{
+    BOOST_ASSERT_MSG(!m_Pointer, "RefMemoryManager::Pool::Reserve() cannot be called after memory acquired");
+    m_Size = std::max(m_Size, numBytes);
+}
+
+void RefMemoryManager::Pool::Acquire()
+{
+    BOOST_ASSERT_MSG(!m_Pointer, "RefMemoryManager::Pool::Acquire() called when memory already acquired"); 
+    BOOST_ASSERT(m_Size >= 0);
+    m_Pointer = ::operator new(size_t(m_Size));
+}
+
+void RefMemoryManager::Pool::Release()
+{
+    BOOST_ASSERT_MSG(m_Pointer, "RefMemoryManager::Pool::Release() called when memory not acquired"); 
+    ::operator delete(m_Pointer);
+    m_Pointer = nullptr;
+}
+
+}