Reference guides
Copy
tensorplay.library
Choosing the kind of custom op
Define a user operator; use as a decorator (torch parity). |
|
Define a Triton-backed operator ( |
|
Mark a Triton kernel as launchable from within a |
Extending custom ops (created from Python or C++)
Top-level kernel registration ( |
|
Top-level autograd registration ( |
|
Top-level fake-kernel registration ( |
Low-level APIs
Scoped registration bundle ( |
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?
