Struct TensorBlockPool 

pub struct TensorBlockPool { /* private fields */ }

Pre-allocated memory pool for tensor operations to reduce allocation overhead.

All blocks share a single configured size. The pool supports allocation, deallocation, reservation, defragmentation, and shrink-to-fit.

Implementations

impl TensorBlockPool

pub fn new(config: PoolConfig) -> Self

Create a new block pool, pre-allocating config.initial_blocks blocks.

pub fn allocate(&mut self, owner: &str) -> Result<u64, String>

Allocate a free block, marking it as BlockStatus::Allocated and assigning it to owner.

If no free block is available and allow_growth is true, a new block is created (up to max_blocks). Returns an error if the pool is exhausted.

pub fn deallocate(&mut self, block_id: u64) -> Result<(), String>

Deallocate the block with the given block_id, marking it BlockStatus::Free, clearing its owner, and incrementing its generation.

pub fn reserve(&mut self, block_id: u64, owner: &str) -> Result<(), String>

Reserve a free block for future use.

Transitions a block from BlockStatus::Free to BlockStatus::Reserved.

pub fn release_reservation(&mut self, block_id: u64) -> Result<(), String>

Release a reservation, returning the block to BlockStatus::Free.

pub fn get_block(&self, block_id: u64) -> Option<&MemoryBlock>

Look up a block by ID.

pub fn free_count(&self) -> usize

Number of free blocks.

pub fn allocated_count(&self) -> usize

Number of allocated blocks.

pub fn utilization(&self) -> f64

Fraction of blocks currently allocated (allocated / total). Returns 0.0 if the pool is empty.

pub fn defragment(&mut self) -> usize

Compact the block list: move all BlockStatus::Free blocks to the end. Resets the generation of relocated free blocks to zero.

Returns the number of free blocks that were relocated.

pub fn shrink_to_fit(&mut self) -> usize

Remove trailing BlockStatus::Free blocks from the pool, shrinking it. Returns the number of blocks removed.

pub fn block_stats(&self) -> BlockPoolStats

Return a snapshot of pool statistics.