Struct TensorKernelRegistry 

pub struct TensorKernelRegistry { /* private fields */ }

Registry of computational kernels with dynamic lookup and versioning.

Implementations

impl TensorKernelRegistry

pub fn new() -> Self

Create an empty registry.

pub fn register( &mut self, name: String, version: u32, precision: KernelPrecision, target: KernelTarget, flops_per_element: f64, tags: Vec<String>, ) -> u64

Register a new kernel and return its assigned kernel_id.

Arguments

• name — Operation name (e.g. "matmul").
• version — Caller-supplied version number.
• precision — Numeric precision.
• target — Execution target.
• flops_per_element— Approximate FLOPs per output element.
• tags — Searchable tag strings.

pub fn lookup(&mut self, query: &KernelQuery) -> Vec<&KernelDescriptor>

Query the registry and return matching KernelDescriptor references.

All non-None query fields must match. Results are sorted by (name asc, version desc, kernel_id asc).

Increments KernelRegistryStats::total_lookups.

pub fn best_for( &mut self, name: &str, precision: KernelPrecision, ) -> Option<&KernelDescriptor>

Select the best kernel for name + precision.

Target preference: Simd > Cpu > Gpu > Generic. Within the same target, the highest version wins.

Increments KernelRegistryStats::total_lookups.

pub fn remove(&mut self, kernel_id: u64) -> bool

Remove a kernel by id.

Returns true if the kernel existed and was removed, false otherwise.

pub fn get(&self, kernel_id: u64) -> Option<&KernelDescriptor>

Retrieve a kernel by id without modifying stats.

pub fn stats(&self) -> &KernelRegistryStats

Return a reference to the current registry statistics.

Trait Implementations

impl Default for TensorKernelRegistry

fn default() -> Self

Returns the “default value” for a type.