Enum ComputeDevice 

pub enum ComputeDevice {
    Cpu {
        device_id: usize,
    },
    Cuda {
        device_id: usize,
    },
    Rocm {
        device_id: usize,
    },
}

Compute device that can execute tensor operations.

Unlike LogicalMemorySpace, which describes where data resides, ComputeDevice identifies the hardware that performs the computation. Multiple compute devices may share access to the same memory space.

Variants

Cpu

CPU compute device.

Fields

device_id: usize

Zero-based CPU device index (0 = default global thread pool).

Cuda

NVIDIA CUDA compute device.

Fields

device_id: usize

Zero-based CUDA device index.

Rocm

AMD ROCm compute device.

Fields

device_id: usize

Zero-based ROCm device index.

Trait Implementations

impl Clone for ComputeDevice

fn clone(&self) -> ComputeDevice

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ComputeDevice

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

Formats the value using the given formatter.

impl Display for ComputeDevice

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

Formats the value using the given formatter.

impl Hash for ComputeDevice

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 ComputeDevice

fn eq(&self, other: &ComputeDevice) -> 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 ComputeDevice

impl Eq for ComputeDevice

impl StructuralPartialEq for ComputeDevice