RefMemoryManager.hpp

Go to the documentation of this file.

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
 
#include <armnn/backends/IMemoryManager.hpp>
 
#include <forward_list>
#include <vector>
 
namespace armnn
{
 
// An implementation of IMemoryManager to be used with RefTensorHandle
class RefMemoryManager : public IMemoryManager
{
public:
    RefMemoryManager();
    virtual ~RefMemoryManager();
 
    class Pool;
 
    Pool* Manage(unsigned int numBytes);
 
    void Allocate(Pool *pool);
 
    void* GetPointer(Pool *pool);
 
    void Acquire() override;
    void Release() override;
 
    class Pool
    {
    public:
        Pool(unsigned int numBytes);
        ~Pool();
 
        void Acquire();
        void Release();
 
        void* GetPointer();
 
        void Reserve(unsigned int numBytes);
 
    private:
        unsigned int m_Size;
        void* m_Pointer;
    };
    
private:
    RefMemoryManager(const RefMemoryManager&) = delete; // Noncopyable
    RefMemoryManager& operator=(const RefMemoryManager&) = delete; // Noncopyable
 
    std::forward_list<Pool> m_Pools;
    std::vector<Pool*> m_FreePools;
};
 
}