tensorplay.library.get_kernel

tensorplay.library.get_kernel(op: str | CustomOpDef, dispatch_key: str) → Callable[[...], Any][source]

dispatch_key accepts "cpu"/"cuda"/"default" and the composite spellings. Raises LookupError when nothing usable is registered (a disabled concrete kernel counts as absent, matching CustomOpDef.set_kernel_enabled() visibility).