Enum Device 

pub enum Device {
    CPU,
    CUDA(usize),
    Metal(usize),
    ROCm(usize),
    WebGPU,
}

Device specification for tensor operations

Variants

CPU

CPU execution

CUDA(usize)

NVIDIA CUDA GPU (with device ID)

Metal(usize)

Apple Metal GPU (with device ID)

ROCm(usize)

AMD ROCm GPU (with device ID)

WebGPU

WebGPU

Implementations

impl Device

pub fn is_gpu(&self) -> bool

Returns true if this device is a GPU

pub fn is_metal(&self) -> bool

Returns true if this device is Metal GPU

pub fn is_cuda(&self) -> bool

Returns true if this device is CUDA GPU

pub fn is_cpu(&self) -> bool

Returns true if this device is CPU

pub fn device_id(&self) -> Option<usize>

Returns the device ID for GPU devices

pub fn metal_if_available(_device_id: usize) -> Device

pub fn cuda_if_available(_device_id: usize) -> Device

pub fn best_available() -> Device

Get the best available device (prefers GPU over CPU)

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

Formats the value using the given formatter.

impl Default for Device

fn default() -> Device

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for Device

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Device

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

Formats the value using the given formatter.

impl Hash for Device

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

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 Serialize for Device

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for Device

impl Eq for Device

impl StructuralPartialEq for Device