Trait GPUArray

pub trait GPUArray: ArrayProtocol {
    // Required methods
    fn to_gpu(&self) -> CoreResult<Box<dyn GPUArray>>;
    fn to_cpu(&self) -> CoreResult<Box<dyn ArrayProtocol>>;
    fn is_on_gpu(&self) -> bool;
    fn device_info(&self) -> HashMap<String, String>;
}

Trait for arrays that can support GPU operations.

Required Methods

fn to_gpu(&self) -> CoreResult<Box<dyn GPUArray>>

Move the array to GPU.

fn to_cpu(&self) -> CoreResult<Box<dyn ArrayProtocol>>

Move the array from GPU to CPU.

fn is_on_gpu(&self) -> bool

Check if the array is on GPU.

fn device_info(&self) -> HashMap<String, String>

Get information about the GPU device that holds this array.

Implementors

impl<T, D> GPUArray for GPUNdarray<T, D>

where T: Clone + Send + Sync + 'static + Zero + Div<f64, Output = T> + Mul<Output = T> + Add<Output = T>, D: Dimension + Clone + Send + Sync + 'static + RemoveAxis,

impl<T: Clone + Send + Sync + 'static> GPUArray for MockGPUArray<T>