Skip to main content

KernelSku

Struct KernelSku 

Source
pub struct KernelSku {
    pub category: OpCategory,
    pub op: u16,
    pub element: ElementKind,
    pub aux_element: Option<ElementKind>,
    pub layout: Option<LayoutSku>,
    pub epilogue: Option<EpilogueKind>,
    pub arch: ArchSku,
    pub backend: BackendKind,
    pub precision_guarantee: PrecisionGuarantee,
}
Expand description

Generalized kernel SKU — covers every op category.

Replaces / generalizes the GEMM-specific GemmSku used by Phase 1/2 plan types. Phase 3+ plan types (UnaryPlan, BinaryPlan, TernaryPlan, …) populate and return this struct from their sku() accessor.

The op field is a category-local discriminant — its interpretation depends on category. For OpCategory::BinaryElementwise it’s a BinaryKind as u16; for OpCategory::UnaryElementwise it’s a UnaryKind as u16; etc. Surfacing it as a flat u16 (rather than a per-category enum) keeps the struct shape stable across categories so it can be hashed into autotuner caches uniformly.

Fields§

§category: OpCategory

Op category. Primary axis for telemetry / selector dispatch.

§op: u16

Category-local op discriminant. Interpret against the per-category *Kind enum that the category’s Plan type uses.

§element: ElementKind

Primary element type the kernel operates on.

§aux_element: Option<ElementKind>

Auxiliary element type, when meaningful (bias element for GEMM bias epilogues, index element for gather / scatter, output cast type for ops that produce a different dtype than the input). None when the op has no auxiliary element.

§layout: Option<LayoutSku>

Layout discriminant for matrix-multiply-shaped kernels. None for op categories that don’t have a row/col layout dimension (elementwise, reduce, scan, …).

§epilogue: Option<EpilogueKind>

Epilogue discriminant for matrix-multiply-shaped kernels. None for op categories that don’t fuse a GEMM epilogue chain.

§arch: ArchSku

Compute capability the selected kernel was compiled for.

§backend: BackendKind

Which underlying compute path served this SKU.

§precision_guarantee: PrecisionGuarantee

Numerical guarantees this SKU provides.

Trait Implementations§

Source§

impl Clone for KernelSku

Source§

fn clone(&self) -> KernelSku

Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§

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

Performs copy-assignment from source. Read more
Source§

impl Copy for KernelSku

Source§

impl Debug for KernelSku

Source§

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

Formats the value using the given formatter. Read more
Source§

impl Eq for KernelSku

Source§

impl Hash for KernelSku

Source§

fn hash<__H>(&self, state: &mut __H)
where __H: Hasher,

Feeds this value into the given Hasher. Read more
1.3.0 · Source§

fn hash_slice<H>(data: &[Self], state: &mut H)
where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher. Read more
Source§

impl PartialEq for KernelSku

Source§

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

Tests for self and other values to be equal, and is used by ==.
1.0.0 (const: unstable) · Source§

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.
Source§

impl StructuralPartialEq for KernelSku

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.