vibespatial.runtime.kernel_registry

Attributes

KernelCallable
KERNEL_VARIANTS

Classes

KernelVariantSpec

Functions

get_kernel_variants(→ tuple[KernelVariantSpec, Ellipsis])
register_kernel_variant(, supports_mixed, ...)	Track declared variants and their dispatch metadata for adaptive planning.

Module Contents

vibespatial.runtime.kernel_registry.KernelCallable

vibespatial.runtime.kernel_registry.KERNEL_VARIANTS: dict[str, list[KernelVariantSpec]]

class vibespatial.runtime.kernel_registry.KernelVariantSpec

kernel_name: str

variant: str

qualified_name: str

kernel_class: vibespatial.runtime.precision.KernelClass | None

execution_modes: tuple[vibespatial.runtime._runtime.ExecutionMode, Ellipsis]

geometry_families: tuple[str, Ellipsis]

supports_mixed: bool

preferred_residency: vibespatial.runtime.residency.Residency | None

precision_modes: tuple[vibespatial.runtime.precision.PrecisionMode, Ellipsis]

min_rows: int | None

max_rows: int | None

tags: tuple[str, Ellipsis]

vibespatial.runtime.kernel_registry.get_kernel_variants(kernel_name: str) → tuple[KernelVariantSpec, Ellipsis]

vibespatial.runtime.kernel_registry.register_kernel_variant(kernel_name: str, variant: str, *, kernel_class: vibespatial.runtime.precision.KernelClass | str | None = None, execution_modes: tuple[vibespatial.runtime._runtime.ExecutionMode | str, Ellipsis] | None = None, geometry_families: tuple[str, Ellipsis] = (), supports_mixed: bool = True, preferred_residency: vibespatial.runtime.residency.Residency | str | None = None, precision_modes: tuple[vibespatial.runtime.precision.PrecisionMode | str, Ellipsis] = (PrecisionMode.AUTO, PrecisionMode.FP32, PrecisionMode.FP64), min_rows: int | None = None, max_rows: int | None = None, tags: tuple[str, Ellipsis] = ()) → collections.abc.Callable[[KernelCallable], KernelCallable]

Track declared variants and their dispatch metadata for adaptive planning.