Enum GpuVendor 

#[non_exhaustive]
pub enum GpuVendor {
    Nvidia,
    Amd,
}

Which GPU stack a device belongs to.

This is an identity, deliberately separate from the device string a tensor library is handed. ROCm libtorch keeps kCUDA, the c10::cuda namespaces, and RCCL exports the NCCL symbol names, so an AMD device is still addressed as CUDA at the API surface while being Amd here. Vendor drives detection, diagnostics, packaging and feature derivation; the API surface is a different axis.

#[non_exhaustive]: Intel is the next entry, and it is not a free-rider on the CUDA API surface the way AMD is (libtorch has a genuinely distinct XPU device type), so adding it will be a real change at every match site rather than a table row. Matches inside this crate stay exhaustive, which is the point: the compiler enumerates the work.

Variants (Non-exhaustive)

Non-exhaustive enums could have additional variants added in future. Therefore, when matching against variants of non-exhaustive enums, an extra wildcard arm must be added to account for any future variants.

Nvidia

Amd

Implementations

impl GpuVendor

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

Lowercase stable token, as written into .arch metadata and cluster-probe JSON.

pub fn parse(s: &str) -> Option<GpuVendor>

Parse the token produced by GpuVendor::as_str. Case- and whitespace-insensitive; also accepts the stack names users type (cuda, rocm, hip).

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

The cargo feature that selects this vendor’s libtorch link set.

Trait Implementations

impl Clone for GpuVendor

fn clone(&self) -> GpuVendor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for GpuVendor

impl Debug for GpuVendor

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

Formats the value using the given formatter.

impl Display for GpuVendor

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

Formats the value using the given formatter.

impl Eq for GpuVendor

impl Hash for GpuVendor

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 GpuVendor

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for GpuVendor