Skip to main content

register_runtime_kernel

Function register_runtime_kernel 

Source
pub fn register_runtime_kernel(
    id: FusedOpId,
    dtypes: &[DType],
    backend: BackendId,
    kernel: KernelRef,
)
Expand description

Bind a kernel for a runtime fused op (id >= RUNTIME_FUSED_BASE) into the global binding table under BindingKey::RuntimeFused(id). dtypes is the per-operand tuple (inputs in order, then output) — the same key shape every static registration uses.