Enum LogicalMemorySpace 

pub enum LogicalMemorySpace {
    MainMemory,
    PinnedMemory,
    GpuMemory {
        device_id: usize,
    },
    ManagedMemory,
}

Logical memory space where tensor data resides.

Separates the concept of “where data lives” from “which hardware computes on it”. A tensor on MainMemory can be processed by any CPU, while a tensor on GpuMemory can be processed by any compute device with access to that GPU memory space.

The variants align with DLPack DLDeviceType constants:

Variant	DLPack device_type
MainMemory	kDLCPU (1)
PinnedMemory	kDLCUDAHost (3) / kDLROCMHost (11)
GpuMemory	kDLCUDA (2) / kDLROCM (10)
ManagedMemory	kDLCUDAManaged (13)

Variants

MainMemory

System main memory (CPU-accessible RAM).

Corresponds to DLPack kDLCPU (device_id = 0).

PinnedMemory

Host memory pinned for fast GPU transfer (cudaMallocHost / hipMallocHost).

Accessible from the CPU but optimized for DMA transfer to/from a specific GPU. Corresponds to DLPack kDLCUDAHost or kDLROCMHost (device_id = 0).

GpuMemory

GPU-resident memory identified by a device ID.

Corresponds to DLPack kDLCUDA or kDLROCM with the given device_id.

Fields

device_id: usize

Zero-based GPU device index (matches DLPack device_id).

ManagedMemory

CUDA Unified/Managed memory (cudaMallocManaged).

Accessible from both CPU and all GPUs. The CUDA runtime handles page migration automatically. Corresponds to DLPack kDLCUDAManaged (device_id = 0).

Trait Implementations

impl Clone for LogicalMemorySpace

fn clone(&self) -> LogicalMemorySpace

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LogicalMemorySpace

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

Formats the value using the given formatter.

impl Hash for LogicalMemorySpace

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for LogicalMemorySpace

fn eq(&self, other: &LogicalMemorySpace) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for LogicalMemorySpace

impl Eq for LogicalMemorySpace

impl StructuralPartialEq for LogicalMemorySpace