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).

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.library.define

Creates the CustomOpDef if absent (kernels are then attached with impl() or Library("ns", "IMPL").impl ). tags is Like Library.define , a full "ns::op(Tensor) -> Tensor" string may be pasted as qualname .

tensorplay.library.impl

types accepts concrete devices ( "CPU" / "CUDA" ) or composite spellings ( CompositeExplicitAutograd → the device-agnostic slot).

tensorplay.library.get_kernel

Help improve this page

On this page