Enum DeviceIndex 

pub enum DeviceIndex {
    Specified(usize),
    Default,
}

Selector for the hardware index of a backend whose devices are simply indexed (e.g. CUDA, ROCm).

Backend factory methods that take an index (Device::cuda, Device::rocm, Device::libtorch_cuda) accept impl Into<DeviceIndex>, so the common shorthand is to pass a plain integer literal:

Device::cuda(0);                    // hardware index 0
Device::cuda(DeviceIndex::Default); // backend-chosen default

Variants

Specified(usize)

Target a specific hardware device by its index.

Default

Let the backend pick its default device (typically index 0).

Implementations

impl DeviceIndex

pub fn new(index: impl Into<usize>) -> Self

Construct a DeviceIndex::Specified from anything convertible into a usize.

Trait Implementations

impl Clone for DeviceIndex

fn clone(&self) -> DeviceIndex

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for DeviceIndex

impl Debug for DeviceIndex

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

Formats the value using the given formatter.

impl Default for DeviceIndex

fn default() -> DeviceIndex

Returns the “default value” for a type.

impl Eq for DeviceIndex

impl From<i32> for DeviceIndex

fn from(i: i32) -> Self

Converts to this type from the input type.

impl From<i64> for DeviceIndex

fn from(i: i64) -> Self

Converts to this type from the input type.

impl From<u32> for DeviceIndex

fn from(i: u32) -> Self

Converts to this type from the input type.

impl From<u64> for DeviceIndex

fn from(i: u64) -> Self

Converts to this type from the input type.

impl From<usize> for DeviceIndex

fn from(i: usize) -> Self

Converts to this type from the input type.

impl Hash for DeviceIndex

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 DeviceIndex

fn eq(&self, other: &DeviceIndex) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for DeviceIndex