Struct MemoryTracker 

pub struct MemoryTracker { /* private fields */ }

Tracks tensor memory allocations for profiling.

Call [alloc] when a tensor is created and [dealloc] when freed.

Example

use shrew::profiler::MemoryTracker;

let mut mem = MemoryTracker::new();
mem.alloc("weight", 1024 * 4);
mem.alloc("bias", 128 * 4);
assert_eq!(mem.current_bytes(), 1024 * 4 + 128 * 4);
mem.dealloc("bias");
assert_eq!(mem.current_bytes(), 1024 * 4);

Implementations

impl MemoryTracker

pub fn new() -> Self

Create a new tracker with zero allocations.

pub fn alloc(&mut self, name: &str, bytes: usize)

Record a tensor allocation.

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

Record a tensor deallocation.

pub fn current_bytes(&self) -> usize

Current live memory in bytes.

pub fn peak_bytes(&self) -> usize

Peak memory usage in bytes.

pub fn total_allocated(&self) -> usize

Total bytes allocated over the tracker’s lifetime.

pub fn alloc_count(&self) -> usize

Number of allocations recorded.

pub fn dealloc_count(&self) -> usize

Number of deallocations recorded.

pub fn live_count(&self) -> usize

Number of currently live allocations.

pub fn reset(&mut self)

Reset the tracker.

pub fn summary(&self) -> String

Format a human-readable memory summary.

Trait Implementations

impl Default for MemoryTracker

fn default() -> Self

Returns the “default value” for a type.