Enum TemplateArg 

pub enum TemplateArg {
    Dtype(Dtype),
    Int(i32),
    Bool(bool),
}

A compile-time template parameter for a MetalKernel.

Each argument is emitted as a template parameter of the generated [[kernel]] function, so the kernel source can refer to it by name (for example T, N or USE_BIAS).

Variants

Dtype(Dtype)

A type parameter, emitted as the matching Metal type (for example float).

Int(i32)

An int parameter.

Bool(bool)

A bool parameter.

Trait Implementations

impl Clone for TemplateArg

fn clone(&self) -> TemplateArg

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for TemplateArg

impl Debug for TemplateArg

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

Formats the value using the given formatter.

impl Eq for TemplateArg

impl PartialEq for TemplateArg

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for TemplateArg