Struct burn_compute::memory_management::SimpleMemoryManagement

pub struct SimpleMemoryManagement<Storage> { /* private fields */ }

Reserves and keeps track of chunks of memory in the storage, and slices upon these chunks.

Implementations

impl<Storage: ComputeStorage> SimpleMemoryManagement<Storage>

pub fn new( storage: Storage, dealloc_strategy: DeallocStrategy, slice_strategy: SliceStrategy ) -> Self

Creates a new instance using the given storage, deallocation strategy and slice strategy.

Trait Implementations

impl<Storage> Debug for SimpleMemoryManagement<Storage>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<Storage: ComputeStorage> MemoryManagement<Storage> for SimpleMemoryManagement<Storage>

fn get(&mut self, handle: &Self::Handle) -> Storage::Resource

Returns the resource from the storage, for the specified handle.

fn reserve(&mut self, size: usize) -> Self::Handle

Reserves memory of specified size using the reserve algorithm, and return a handle to the reserved memory.

Also clean ups, removing unused slices, and chunks if permitted by deallocation strategy.

type Handle = SimpleHandle

The associated type Handle must implement MemoryHandle

fn alloc(&mut self, size: usize) -> Self::Handle

Bypass the memory allocation algorithm to allocate data directly.

fn dealloc(&mut self, handle: &Self::Handle)

Bypass the memory allocation algorithm to deallocate data directly.

fn storage(&mut self) -> &mut Storage

Fetch the storage used by the memory manager.