Struct CrossDeviceMemoryManager 

pub struct CrossDeviceMemoryManager { /* private fields */ }

Cross-device memory manager

Implementations

impl CrossDeviceMemoryManager

pub fn new() -> Self

Create a new cross-device memory manager

pub fn register_device(&self, device: Arc<dyn Device>) -> CoreResult<()>

Register a device with the manager

pub fn set_default_device(&self, devicetype: DeviceType) -> CoreResult<()>

Set the default device

pub fn get_default_device(&self) -> Option<DeviceType>

Get the default device

pub fn allocate<T: 'static>( self: &Arc<Self>, device_type: &DeviceType, count: usize, ) -> CoreResult<CrossDeviceBuffer<T>>

Allocate memory on a specific device

pub fn allocate_default<T: 'static>( self: &Arc<Self>, count: usize, ) -> CoreResult<CrossDeviceBuffer<T>>

Allocate memory on the default device

pub fn transfer<T: 'static + Copy>( self: &Arc<Self>, src_buffer: &CrossDeviceBuffer<T>, dst_device: &DeviceType, ) -> CoreResult<CrossDeviceBuffer<T>>

Transfer data between devices

pub fn synchronize_all(&self) -> CoreResult<()>

Synchronize all devices

pub fn get_memory_statistics(&self) -> MemoryStatistics

Get memory statistics

pub fn cleanup_unused_allocations(&self, maxage: Duration) -> usize

Clean up unused allocations

Trait Implementations

impl Default for CrossDeviceMemoryManager

fn default() -> Self

Returns the “default value” for a type.