Struct NDArrayPool 

pub struct NDArrayPool { /* private fields */ }

NDArray factory with free-list reuse and memory tracking.

Mimics C++ ADCore’s NDArrayPool: on alloc, checks the free list for a buffer with sufficient capacity. On release, returns the buffer to the free list for future reuse. The free list is sorted by capacity (descending) and excess entries are dropped when max_memory is exceeded.

Implementations

impl NDArrayPool

pub fn new(max_memory: usize) -> Self

pub fn alloc( &self, dims: Vec<NDDimension>, data_type: NDDataType, ) -> ADResult<NDArray>

Allocate an NDArray. Tries to reuse a free-list entry with sufficient capacity.

pub fn alloc_copy(&self, source: &NDArray) -> ADResult<NDArray>

Allocate a copy of an existing NDArray (new unique_id, data cloned).

pub fn release(&self, array: NDArray)

Return an array to the free list for future reuse.

pub fn empty_free_list(&self)

Clear all entries from the free list.

pub fn allocated_bytes(&self) -> u64

pub fn num_alloc_buffers(&self) -> u32

pub fn num_free_buffers(&self) -> u32

pub fn max_memory(&self) -> usize

pub fn alloc_handle( pool: &Arc<Self>, dims: Vec<NDDimension>, data_type: NDDataType, ) -> ADResult<NDArrayHandle>

Allocate an NDArray wrapped in a pool-aware handle. On final drop, the array is returned to this pool’s free list.