Enum autograph::Device

pub enum Device {
    Cpu(Arc<Cpu>),
}

Device is an enum that is used to select whether to store and execute operations on the cpu or a gpu.

You can use the From trait to create a device from a Cpu or CudaGpu:

let cpu = Device::from(Cpu::new());

Device can be cloned, which copies the pointer. Each Tensor will have a copy of the Device so that it can execute operations.

Variants

Cpu(Arc<Cpu>)

Implementations

impl Device

pub fn synchronize(&self)

For cpu does nothing. For cuda, blocks until all operations on the device are finished. Only necessary for timing ie for benchmarks. Any tranfers back to the cpu will implicitly synchronize.

Trait Implementations

impl Clone for Device

fn clone(&self) -> Device

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Device

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

Formats the value using the given formatter.

impl From<Arc<Cpu>> for Device

fn from(cpu: Arc<Cpu>) -> Self

Performs the conversion.

impl PartialEq<Device> for Device

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

This method tests for self and other values to be equal, and is used by ==.

#[must_use]fn ne(&self, other: &Rhs) -> bool

This method tests for !=.