Skip to content

Commit 914dbc5

Browse files
committed
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]
1 parent fbf38a1 commit 914dbc5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

runtime/kernel/test/make_boxed_from_unboxed_functor_test.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ TEST_F(MakeBoxedFromUnboxedFunctorTest, Basic) {
4040
Kernel my_kernel =
4141
Kernel::make_boxed_kernel("my_ns::my_op.out", EXECUTORCH_FN(my_op_out));
4242
ArrayRef<Kernel> kernels_array = ArrayRef<Kernel>(my_kernel);
43-
// patternlint-disable-next-line clang-diagnostic-unused-variable
43+
// @lint-ignore CLANGTIDY
4444
auto s1 = register_kernels(kernels_array);
4545
EXPECT_TRUE(hasOpsFn("my_ns::my_op.out"));
4646
}
@@ -49,7 +49,7 @@ TEST_F(MakeBoxedFromUnboxedFunctorTest, UnboxLogicWorks) {
4949
Kernel my_kernel =
5050
Kernel::make_boxed_kernel("my_ns::set_1.out", EXECUTORCH_FN(set_1_out));
5151
ArrayRef<Kernel> kernels_array = ArrayRef<Kernel>(my_kernel);
52-
// patternlint-disable-next-line clang-diagnostic-unused-variable
52+
// @lint-ignore CLANGTIDY
5353
auto s1 = register_kernels(kernels_array);
5454
EXPECT_TRUE(hasOpsFn("my_ns::set_1.out"));
5555

0 commit comments

Comments
 (0)