Struct MemoryStats 

pub struct MemoryStats {
    pub total_allocated: usize,
    pub tensor_count: usize,
    pub memory_by_name: HashMap<String, usize>,
}

Memory usage tracking for GPU tensors

Fields

total_allocated: usize

Total memory allocated (bytes)

tensor_count: usize

Number of tensors tracked

memory_by_name: HashMap<String, usize>

Memory by tensor name

Implementations

impl MemoryStats

pub fn new() -> Self

Create a new memory stats tracker

pub fn track_tensor(&mut self, name: String, tensor: &Tensor)

Track a tensor’s memory usage

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

Untrack a tensor

pub fn total_bytes(&self) -> usize

Get total allocated memory in bytes

pub fn total_mb(&self) -> f64

Get total allocated memory in MB

pub fn total_gb(&self) -> f64

Get total allocated memory in GB

pub fn clear(&mut self)

Clear all tracked tensors

Trait Implementations

impl Clone for MemoryStats

fn clone(&self) -> MemoryStats

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MemoryStats

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

Formats the value using the given formatter.

impl Default for MemoryStats

fn default() -> Self

Returns the “default value” for a type.