luci_interpreter::TestMemoryManager Class Reference

#include <TestMemoryManager.h>

Collaboration diagram for luci_interpreter::TestMemoryManager:

Public Member Functions
void	allocate_memory (luci_interpreter::Tensor &tensor) final
void	release_memory (luci_interpreter::Tensor &tensor) final
	~TestMemoryManager () override
Public Member Functions inherited from luci_interpreter::IMemoryManager
virtual	~IMemoryManager ()=default

Detailed Description

Definition at line 27 of file TestMemoryManager.h.

Constructor & Destructor Documentation

~TestMemoryManager()

luci_interpreter::TestMemoryManager::~TestMemoryManager ( )

inlineoverride

Definition at line 33 of file TestMemoryManager.h.

  {
    for (auto allocation : allocations)
    {
      delete[] allocation;
    }
  }

Member Function Documentation

allocate_memory()

void luci_interpreter::TestMemoryManager::allocate_memory ( luci_interpreter::Tensor &  tensor )

finalvirtual

Implements luci_interpreter::IMemoryManager.

Definition at line 22 of file TestMemoryManager.cpp.

{
  if (!tensor.is_allocatable())
  {
    return;
  }
  if (tensor.is_data_allocated())
  {
    release_memory(tensor);
  }
  const auto element_size = getDataTypeSize(tensor.element_type());
  const auto num_elements = tensor.shape().num_elements();
 
  auto *data = new uint8_t[num_elements * element_size];
  allocations.push_back(data);
  tensor.set_data_buffer(data);
}

References flatbuffers::data(), luci_interpreter::getDataTypeSize(), luci::must_cast(), and release_memory().

release_memory()

void luci_interpreter::TestMemoryManager::release_memory ( luci_interpreter::Tensor &  tensor )

finalvirtual

Implements luci_interpreter::IMemoryManager.

Definition at line 40 of file TestMemoryManager.cpp.

{
  tensor.set_data_buffer(nullptr);
}

Referenced by allocate_memory().

---

The documentation for this class was generated from the following files:

• compiler/luci-interpreter/include/luci_interpreter/TestMemoryManager.h
• compiler/luci-interpreter/src/TestMemoryManager.cpp