#include <MemoryManager.h>

Public Member Functions
	LayerScopeMemoryManager ()

void	allocate (void)

uint8_t *	getBuffer (const LayerScopeTensorIndex &ind) const

void	deallocate (void)

void	claimPlan (const LayerScopeTensorIndex &ind, uint32_t size)

void	releasePlan (const LayerScopeTensorIndex &ind)

Detailed Description

Definition at line 67 of file MemoryManager.h.

Constructor & Destructor Documentation

◆ LayerScopeMemoryManager()

onert::backend::train::LayerScopeMemoryManager::LayerScopeMemoryManager ( )

Definition at line 92 of file MemoryManager.cc.

                                                 : _mem_planner{createMemoryPlanner()}
{
  // DO NOTHING
}

Member Function Documentation

◆ allocate()

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

Definition at line 103 of file MemoryManager.cc.

{
  _mem_alloc = std::make_shared<basic::Allocator>(_mem_planner->capacity());
  assert(_mem_alloc->base());
}

◆ claimPlan()

void onert::backend::train::LayerScopeMemoryManager::claimPlan	(	const LayerScopeTensorIndex &	ind,
		uint32_t	size
	)

Definition at line 118 of file MemoryManager.cc.

{
  _mem_planner->claim(ind, size);
}

References size.

◆ deallocate()

void onert::backend::train::LayerScopeMemoryManager::deallocate ( void )

Definition at line 116 of file MemoryManager.cc.

116{ _mem_alloc->release(); }

◆ getBuffer()

uint8_t * onert::backend::train::LayerScopeMemoryManager::getBuffer ( const LayerScopeTensorIndex & ind ) const

Definition at line 109 of file MemoryManager.cc.

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

◆ releasePlan()

void onert::backend::train::LayerScopeMemoryManager::releasePlan ( const LayerScopeTensorIndex & ind )

Definition at line 123 of file MemoryManager.cc.

{
  _mem_planner->release(ind);
}

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