#include <MemoryManager.h>

Public Member Functions
	DisposableMemoryManager ()

void	allocate (void)

uint8_t *	getBuffer (const DisposableTensorIndex &ind) const

void	deallocate (void)

void	claimPlan (const DisposableTensorIndex &ind, uint32_t size)

void	releasePlan (const DisposableTensorIndex &ind)

std::shared_ptr< basic::Allocator >	getMemAlloc ()

Detailed Description

Definition at line 44 of file MemoryManager.h.

Constructor & Destructor Documentation

◆ DisposableMemoryManager()

onert::backend::train::DisposableMemoryManager::DisposableMemoryManager ( )

Definition at line 52 of file MemoryManager.cc.

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

Member Function Documentation

◆ allocate()

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

Definition at line 79 of file MemoryManager.cc.

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

◆ claimPlan()

void onert::backend::train::DisposableMemoryManager::claimPlan	(	const DisposableTensorIndex &	ind,
		uint32_t	size
	)

Definition at line 69 of file MemoryManager.cc.

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

References size.

◆ deallocate()

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

inline

Definition at line 51 of file MemoryManager.h.

51{ _mem_alloc->release(); }

◆ getBuffer()

uint8_t * onert::backend::train::DisposableMemoryManager::getBuffer ( const DisposableTensorIndex & ind ) const

Definition at line 85 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;
}

◆ getMemAlloc()

std::shared_ptr< basic::Allocator > onert::backend::train::DisposableMemoryManager::getMemAlloc ( )

inline

Definition at line 56 of file MemoryManager.h.

56{ return _mem_alloc; }

◆ releasePlan()

void onert::backend::train::DisposableMemoryManager::releasePlan ( const DisposableTensorIndex & ind )

Definition at line 74 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