Struct MemoryBudgetManager 

pub struct MemoryBudgetManager { /* private fields */ }

Memory budget manager for training.

Implementations

impl MemoryBudgetManager

pub fn new(budget_bytes: usize) -> Self

Create a new memory budget manager.

Arguments

• budget_bytes - Total memory budget in bytes

pub fn from_gb(gb: f64) -> Self

Create from GB specification.

pub fn try_allocate(&mut self, name: &str, bytes: usize) -> bool

Try to allocate memory.

Returns true if allocation succeeded, false if would exceed budget.

pub fn free(&mut self, name: &str)

Free allocated memory.

pub fn available(&self) -> usize

Get available memory.

pub fn utilization(&self) -> f64

Get utilization percentage.

pub fn summary(&self) -> String

Get allocation summary.

Trait Implementations

impl Clone for MemoryBudgetManager

fn clone(&self) -> MemoryBudgetManager

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MemoryBudgetManager

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

Formats the value using the given formatter.