Struct DeviceMemoryManager 

pub struct DeviceMemoryManager { /* private fields */ }

Device memory manager for cross-device operations

Implementations

impl DeviceMemoryManager

pub fn new(max_cachesize: usize) -> Result<Self, CoreError>

Create a new device memory manager

pub fn is_device_available(&self, device: DeviceType) -> bool

Check if a device type is available

pub fn available_devices(&self) -> Vec<DeviceType>

Get a list of available devices

pub fn transfer_to_device<T, S, D>( &self, array: &ArrayBase<S, D>, device: DeviceType, options: Option<TransferOptions>, ) -> CoreResult<DeviceArray<T, D>>

where T: GpuDataType, S: RawData<Elem = T> + Data, D: Dimension,

Transfer data from host to device

pub fn transfer_to_host<T, D>( &self, devicearray: &DeviceArray<T, D>, options: Option<TransferOptions>, ) -> CoreResult<Array<T, D>>

where T: GpuDataType, D: Dimension,

Transfer data from device to host

pub fn transfer_between_devices<T, D>( &self, devicearray: &DeviceArray<T, D>, target_device: DeviceType, options: Option<TransferOptions>, ) -> CoreResult<DeviceArray<T, D>>

where T: GpuDataType, D: Dimension,

Transfer data between devices

pub fn clear_cache(&self)

Clear the cache

pub fn execute_kernel<T, D>( &self, devicearray: &DeviceArray<T, D>, kernel_name: &str, params: HashMap<String, KernelParam>, ) -> CoreResult<()>

where T: GpuDataType, D: Dimension,

Execute a kernel on a device array

Trait Implementations

impl Debug for DeviceMemoryManager

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

Formats the value using the given formatter.