Skip to content

Commit

Permalink
Unbreak custom_kernel_example test (#7985)
Browse files Browse the repository at this point in the history
This test doesn't run in OSS CI (#7984), so I missed it.

Test Plan: ran relevant test internally with buck2
swolchok authored Jan 28, 2025
1 parent b57a292 commit 533c1aa
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion kernels/test/custom_kernel_example/op_relu.cpp
Original file line number Diff line number Diff line change
@@ -79,7 +79,7 @@ my_relu_out(KernelRuntimeContext& context, const Tensor& input, Tensor& out) {
break;

switch (input.scalar_type()) {
ET_FORALL_REAL_TYPES(RELU)
ET_FORALL_REALHBF16_TYPES(RELU)
default:
ET_KERNEL_CHECK_MSG(
context,

0 comments on commit 533c1aa

Please sign in to comment.