Enum DeviceSpec 

pub enum DeviceSpec {
    Cpu,
    Cuda {
        device_id: usize,
    },
    Metal {
        device_id: usize,
    },
    WebGpu,
}

Device specification parsed from a device string.

This enum represents different compute devices that can execute kernels. It’s used throughout the compilation pipeline for device selection and kernel caching.

Variants

Cpu

CPU device (single-threaded or multi-threaded execution)

Cuda

CUDA GPU device with specific device ID

Fields

device_id: usize

Metal

Metal GPU device (Apple Silicon) with specific device ID

Fields

device_id: usize

WebGpu

WebGPU device (browser or native WebGPU)

Implementations

impl DeviceSpec

pub fn canonicalize(&self) -> String

Canonicalize the device spec to a standard string representation.

Examples

use morok_dtype::DeviceSpec;

assert_eq!(DeviceSpec::Cpu.canonicalize(), "CPU");
assert_eq!(DeviceSpec::Cuda { device_id: 0 }.canonicalize(), "CUDA:0");
assert_eq!(DeviceSpec::Cuda { device_id: 1 }.canonicalize(), "CUDA:1");

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

Get maximum buffer count for this device.

Returns None if the device has no buffer limit (effectively unlimited).

Known limits:

• Metal: 31 buffers (Apple Silicon hardware limit)
• WebGPU: 8 buffers (WebGPU specification limit)
• CPU/CUDA: None (no practical limit)

pub fn base_type(&self) -> &'static str

Get the base device type string (strips device ID).

Used for device factory lookup and cache key construction. Unlike canonicalize(), this returns a static string without device ID.

Examples

use morok_dtype::DeviceSpec;

assert_eq!(DeviceSpec::Cpu.base_type(), "CPU");
assert_eq!(DeviceSpec::Cuda { device_id: 0 }.base_type(), "CUDA");
assert_eq!(DeviceSpec::Cuda { device_id: 1 }.base_type(), "CUDA");

Trait Implementations

impl Clone for DeviceSpec

fn clone(&self) -> DeviceSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DeviceSpec

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for DeviceSpec

fn deserialize<__D>( __deserializer: __D, ) -> Result<DeviceSpec, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl DeviceSpecExt for DeviceSpec

fn parse(s: &str) -> Result<DeviceSpec, Error>

Parse a device string into a DeviceSpec.

impl Hash for DeviceSpec

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 DeviceSpec

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

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Eq for DeviceSpec

impl StructuralPartialEq for DeviceSpec