#include <MemoryManager.h>

Collaboration diagram for onert::backend::train::TrainableMemoryManager:

Public Member Functions
	TrainableMemoryManager (uint32_t optimizer_vars_count)

virtual	~TrainableMemoryManager ()=default

void	allocate (void)

uint8_t *	getOptVarBuffer (const ir::OperandIndex &ind, uint32_t pos_var) const

Public Member Functions inherited from onert::backend::basic::MemoryManager
	MemoryManager ()

	MemoryManager (const std::string)

virtual	~MemoryManager ()=default

void	allocate (void)

uint8_t *	getBuffer (const ir::OperandIndex &ind) const

void	deallocate (void)

void	claimPlan (const ir::OperandIndex &ind, uint32_t size)

void	releasePlan (const ir::OperandIndex &ind)

Additional Inherited Members
Protected Attributes inherited from onert::backend::basic::MemoryManager
std::unordered_map< ir::OperandIndex, Block >	_tensor_mem_map

std::shared_ptr< IMemoryPlanner< ir::OperandIndex > >	_mem_planner

std::shared_ptr< Allocator >	_mem_alloc

Detailed Description

Definition at line 30 of file MemoryManager.h.

Constructor & Destructor Documentation

◆ TrainableMemoryManager()

onert::backend::train::TrainableMemoryManager::TrainableMemoryManager ( uint32_t optimizer_vars_count )

Definition at line 28 of file MemoryManager.cc.

  : _optim_vars_count{optim_vars_count}
{
  // DO NOTHING
}

◆ ~TrainableMemoryManager()

virtual onert::backend::train::TrainableMemoryManager::~TrainableMemoryManager ( )

virtualdefault

Member Function Documentation

◆ allocate()

void onert::backend::train::TrainableMemoryManager::allocate ( void )

Definition at line 34 of file MemoryManager.cc.

{
  _mem_alloc = std::make_shared<basic::Allocator>(_mem_planner->capacity());
  assert(_mem_alloc->base());
 
  const auto vars_capacity = _mem_planner->capacity() * _optim_vars_count;
  _var_mem_alloc = std::make_shared<basic::Allocator>(vars_capacity);
}

References onert::backend::basic::MemoryManager::_mem_alloc, and onert::backend::basic::MemoryManager::_mem_planner.

◆ getOptVarBuffer()

uint8_t * onert::backend::train::TrainableMemoryManager::getOptVarBuffer	(	const ir::OperandIndex &	ind,
		uint32_t	pos_var
	)		const

Definition at line 43 of file MemoryManager.cc.

{
  assert(_mem_planner->memory_plans().find(ind) != _mem_planner->memory_plans().end());
  const auto var_offset = pos_var * _mem_planner->capacity();
  const auto &mem_blk = _mem_planner->memory_plans().at(ind);
  return _var_mem_alloc->base() + var_offset + mem_blk.offset;
}

References onert::backend::basic::MemoryManager::_mem_planner.

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

runtime/onert/backend/train/MemoryManager.h
runtime/onert/backend/train/MemoryManager.cc

Public Member Functions