Skip to content
Closed
Prev Previous commit
Next Next commit
Update on "[kernel] Add template based unboxing"
Adding a new feature to allow users to bypass codegen and register their
kernels directly. This is very useful for custom kernels for custom ops.

Example usage:
```
Tensor& my_op(RuntimeContext& ctx, const Tensor& self, const Tensor& other, Tensor& out) {
    // ...
    return out;
}

Kernel my_kernel = Kernel.make_boxed_kernel("my_ns::my_op",EXECUTORCH_FN(my_op));

register_kernels({my_kernel});
```

Differential Revision: [D51553099](https://our.internmc.facebook.com/intern/diff/D51553099)

[ghstack-poisoned]
  • Loading branch information
larryliu0820 committed Feb 29, 2024
commit c5b2b537dddab6fe572153a4b210fa273b983237

This merge commit was added into this branch cleanly.

There are no new changes to show, but you can still view the diff.