Enum ComputeDevice 

pub enum ComputeDevice {
    Cpu,
    Cuda {
        device_id: usize,
    },
    Wgpu {
        adapter_index: u32,
    },
}

Compute device for training

Variants

Cpu

CPU-only execution

Cuda

CUDA GPU with device ID

Fields

device_id: usize

Wgpu

wgpu GPU with adapter index (Vulkan/Metal/DX12)

Fields

adapter_index: u32

Implementations

impl ComputeDevice

pub fn auto_detect() -> Self

Auto-detect best available device

Priority: CUDA (if ≥6GB) > wgpu (if available) > CPU

pub fn cuda_available() -> bool

Check if CUDA is available

pub fn wgpu_available() -> bool

Check if wgpu GPU is available

pub const fn is_cuda(&self) -> bool

Check if this device is CUDA

pub const fn is_cpu(&self) -> bool

Check if this device is CPU

pub const fn is_wgpu(&self) -> bool

Check if this device is wgpu

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

Get device ID for CUDA devices

pub const fn adapter_index(&self) -> Option<u32>

Get adapter index for wgpu devices

pub fn detect_all_devices() -> Vec<Self>

Detect all available compute devices on this machine.

Enumerates all CUDA GPUs and wgpu adapters. Returns at least one device (CPU fallback if no GPUs found).

Used by distributed training to discover multi-GPU configurations.

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

Formats the value using the given formatter.

impl Default for ComputeDevice

fn default() -> Self

Returns the “default value” for a type.

impl Display for ComputeDevice

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

Formats the value using the given formatter.

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