Struct AutoDevice 

pub struct AutoDevice<T, D>
where
    T: Clone + Send + Sync + 'static + Zero + Div<f64, Output = T> + Default + Mul<Output = T> + Add<Output = T>,
    D: Dimension + RemoveAxis,
    SliceInfo<[SliceInfoElem; 1], Dim<[usize; 1]>, Dim<[usize; 1]>>: SliceArg<D>,
    SliceInfo<[SliceInfoElem; 2], Dim<[usize; 2]>, Dim<[usize; 2]>>: SliceArg<D>,
{ /* private fields */ }

A wrapper for arrays that automatically chooses the best device.

This wrapper automatically places arrays on the most appropriate device based on their size and the operations being performed.

Implementations

impl<T, D> AutoDevice<T, D>

where T: Clone + Send + Sync + 'static + Zero + Div<f64, Output = T> + Default + Mul<Output = T> + Add<Output = T>, D: Dimension + RemoveAxis + 'static, SliceInfo<[SliceInfoElem; 1], Dim<[usize; 1]>, Dim<[usize; 1]>>: SliceArg<D>, SliceInfo<[SliceInfoElem; 2], Dim<[usize; 2]>, Dim<[usize; 2]>>: SliceArg<D>,

pub fn new(array: Array<T, D>) -> Self

Create a new auto-device array.

pub fn on_device(&mut self, device: DeviceType) -> &dyn ArrayProtocol

Get the array on the specified device.

pub fn device(&self) -> DeviceType

Get the current device.

pub const fn array(&self) -> &Array<T, D>

Get the underlying array.

Trait Implementations

impl<T, D> ArrayProtocol for AutoDevice<T, D>

where T: Clone + Send + Sync + 'static + Zero + Div<f64, Output = T> + Default + Mul<Output = T> + Add<Output = T>, D: Dimension + RemoveAxis + 'static, SliceInfo<[SliceInfoElem; 1], Dim<[usize; 1]>, Dim<[usize; 1]>>: SliceArg<D>, SliceInfo<[SliceInfoElem; 2], Dim<[usize; 2]>, Dim<[usize; 2]>>: SliceArg<D>,

fn array_function( &self, func: &ArrayFunction, types: &[TypeId], args: &[Box<dyn Any>], kwargs: &HashMap<String, Box<dyn Any>>, ) -> Result<Box<dyn Any>, NotImplemented>

Implementation of the array protocol.

fn as_any(&self) -> &dyn Any

Get the array as Any for downcasting

fn shape(&self) -> &[usize]

Get the shape of the array (default implementation returns empty slice)

fn dtype(&self) -> TypeId

Get the data type of the array (default implementation returns f64)

fn box_clone(&self) -> Box<dyn ArrayProtocol>

Clone this array protocol object.

impl<T, D> Clone for AutoDevice<T, D>

where T: Clone + Send + Sync + 'static + Zero + Div<f64, Output = T> + Default + Mul<Output = T> + Add<Output = T>, D: Dimension + RemoveAxis + 'static, SliceInfo<[SliceInfoElem; 1], Dim<[usize; 1]>, Dim<[usize; 1]>>: SliceArg<D>, SliceInfo<[SliceInfoElem; 2], Dim<[usize; 2]>, Dim<[usize; 2]>>: SliceArg<D>,

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T, D> Debug for AutoDevice<T, D>

where T: Clone + Send + Sync + Debug + 'static + Zero + Div<f64, Output = T> + Default + Mul<Output = T> + Add<Output = T>, D: Dimension + RemoveAxis + Debug + 'static, SliceInfo<[SliceInfoElem; 1], Dim<[usize; 1]>, Dim<[usize; 1]>>: SliceArg<D>, SliceInfo<[SliceInfoElem; 2], Dim<[usize; 2]>, Dim<[usize; 2]>>: SliceArg<D>,

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

Formats the value using the given formatter.