Enum Device 

pub enum Device {
    Cpu,
    Cuda(usize),
}

Device on which a tensor’s data resides.

Defined in Phase 1 with only Cpu functional. Cuda is present from day one so the type is baked into every API before GPU work begins.

Variants

Cpu

CPU main memory.

Cuda(usize)

CUDA GPU with the given device index.

Implementations

impl Device

pub fn is_cpu(&self) -> bool

Returns true if this is a CPU device.

pub fn is_cuda(&self) -> bool

Returns true if this is a CUDA device.

Trait Implementations

impl Clone for Device

fn clone(&self) -> Device

Returns a duplicate 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<(), Error>

Formats the value using the given formatter.

impl Default for Device

fn default() -> Device

Returns the “default value” for a type.

impl Display for Device

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

Formats the value using the given formatter.

impl Hash for Device

fn hash<__H>(&self, state: &mut __H)

where __H: Hasher,

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for Device

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

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for Device

impl Eq for Device

impl StructuralPartialEq for Device