Struct cubecl_runtime::memory_management::simple::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, binding: Self::Binding) -> Storage::Resource

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

fn reserve<Sync: FnOnce()>(&mut self, size: usize, _sync: Sync) -> 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 that must implement MemoryHandle.

type Binding = SimpleBinding

The associated type that must implement MemoryBinding

fn alloc<Sync: FnOnce()>(&mut self, size: usize, _sync: Sync) -> Self::Handle

Bypass the memory allocation algorithm to allocate data directly.

fn dealloc(&mut self, binding: Self::Binding)

Bypass the memory allocation algorithm to deallocate data directly.

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

Fetch the storage used by the memory manager.