Skip to main content

Module memory_management

Module memory_management 

Source
Expand description

Memory management module.

Modules§

drop_queue
Export utilities to keep track of CPU buffers when performing async data copies.

Structs§

ManagedMemoryBinding
Binding of a memory handle
ManagedMemoryHandle
Managed Memory handle
ManagedMemoryId
Managed memory unique identifier.
MemoryManagement
Reserves and keeps track of chunks of memory in the storage, and slices upon these chunks.
MemoryManagementOptions
The options for creating a new MemoryManagement instance.
MemoryPoolOptions
Options to create a memory pool.
MemoryUsage
Amount of memory in use by this allocator and statistics on how much memory is reserved and wasted in total.

Enums§

MemoryAllocationMode
The mode of allocation used.
MemoryConfiguration
High level configuration of memory management.
PoolType
The type of memory pool to use.

Traits§

MemoryHandle
The managed tensor buffer handle that points to some memory segment. It should not contain actual data.

Functions§

optimal_align
Calculates a best-effort heuristic for the alignment of row-aligned tensors. Prefers contiguous alignments for unit dimensions, 16-byte minimum alignment for non-unit, scaling with input size up to buffer_align.