#include <SimpleMemoryManager.h>

Collaboration diagram for luci_interpreter::SimpleMemoryManager:

Public Member Functions
void	allocate_memory (luci_interpreter::Tensor &tensor) final

void	release_memory (luci_interpreter::Tensor &tensor) final

uint8_t *	allocate_memory (const circle::Tensor *tensor)

void	release_memory (uint8_t *data)

Public Member Functions inherited from luci_interpreter::IMemoryManager
virtual	~IMemoryManager ()=default

Detailed Description

Definition at line 25 of file SimpleMemoryManager.h.

Member Function Documentation

◆ allocate_memory() [1/2]

uint8_t * luci_interpreter::SimpleMemoryManager::allocate_memory ( const circle::Tensor * tensor )

Definition at line 24 of file SimpleMemoryManager.cpp.

{
  const auto element_size = getDataTypeSize(Tensor::element_type(tensor));
  const auto num_elements = Tensor::num_elements(tensor);
 
  assert(element_size * num_elements > 0);
 
  return new uint8_t[num_elements * element_size];
}

References luci_interpreter::getDataTypeSize().

◆ allocate_memory() [2/2]

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

finalvirtual

Implements luci_interpreter::IMemoryManager.

Definition at line 22 of file SimpleMemoryManager.cpp.

{
  if (!tensor.is_allocatable())
  {
    return;
  }
  if (tensor.is_data_allocated())
  {
    release_memory(tensor);
  }
  const auto element_size = getDataTypeSize(tensor.element_type());
 
  // Use large_num_elements to avoid overflow
  const auto num_elements = tensor.shape().large_num_elements();
 
  auto *data = new uint8_t[num_elements * element_size];
  tensor.set_data_buffer(data);
}

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

◆ release_memory() [1/2]

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

finalvirtual

Implements luci_interpreter::IMemoryManager.

Definition at line 41 of file SimpleMemoryManager.cpp.

{
  if (!tensor.is_data_allocated())
  {
    tensor.set_data_buffer(nullptr);
    return;
  }
  auto data = tensor.data<uint8_t>();
  delete[] data;
  tensor.set_data_buffer(nullptr);
}

References flatbuffers::data().

Referenced by allocate_memory().

◆ release_memory() [2/2]

void luci_interpreter::SimpleMemoryManager::release_memory ( uint8_t * data )

Definition at line 34 of file SimpleMemoryManager.cpp.

{
  if (data == nullptr)
    return;
 
  delete[] data;
}