Enum Device 

pub enum Device {
    Cpu,
    Cuda(usize),
    Xpu(usize),
    Mps(usize),
    Meta,
}

Device on which a tensor’s data resides.

Meta is a special device that does not allocate any backing memory: meta tensors carry shape, dtype, and device information but no data. They are useful for shape inference, dry-run model construction, and inspecting parameter counts of huge models without actually allocating the weights. Mirrors torch.device("meta").

Xpu mirrors PyTorch’s torch.device("xpu") and addresses Intel GPUs (Arc series, Data Center GPU Max) via the portable CubeCL wgpu runtime that the ferrotorch-xpu crate wraps. CL-452.

Variants

Cpu

CPU main memory.

Cuda(usize)

CUDA GPU with the given device index.

Xpu(usize)

Intel XPU (Arc / Data Center GPU Max) with the given device index. Accessed via ferrotorch-xpu which wraps a CubeCL wgpu runtime. CL-452.

Mps(usize)

Apple Silicon Metal Performance Shaders. The usize is the Metal device index (0 is the system default GPU). Mirrors torch.device("mps"). Implemented via ferrotorch-mps. (#451)

Meta

Meta device — shape-only, no backing storage. Operations that need data return an error; operations that only manipulate metadata (reshape, view, permute, narrow, transpose, …) work normally and produce meta tensors as output. CL-395.

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.

pub fn is_xpu(&self) -> bool

Returns true if this is an Intel XPU device. CL-452.

pub fn is_mps(&self) -> bool

Returns true if this is an Apple MPS device. (#451)

pub fn is_meta(&self) -> bool

Returns true if this is the meta device (shape-only, no data).

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

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

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