luci_interpreter::SimpleMemoryManager Class Reference

#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(), and luci::must_cast().

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(), luci::must_cast(), 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(), and luci::must_cast().

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;
}

References flatbuffers::data().

---

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

• compiler/luci-interpreter/include/luci_interpreter/SimpleMemoryManager.h
• onert-micro/luci-interpreter/src/memory_managers/SimpleMemoryManager.h
• compiler/luci-interpreter/src/SimpleMemoryManager.cpp
• onert-micro/luci-interpreter/src/memory_managers/SimpleMemoryManager.cpp