Skip to content

Commit 533c1aa

Browse files
authored
Unbreak custom_kernel_example test (#7985)
This test doesn't run in OSS CI (#7984), so I missed it. Test Plan: ran relevant test internally with buck2
1 parent b57a292 commit 533c1aa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kernels/test/custom_kernel_example/op_relu.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ my_relu_out(KernelRuntimeContext& context, const Tensor& input, Tensor& out) {
7979
break;
8080

8181
switch (input.scalar_type()) {
82-
ET_FORALL_REAL_TYPES(RELU)
82+
ET_FORALL_REALHBF16_TYPES(RELU)
8383
default:
8484
ET_KERNEL_CHECK_MSG(
8585
context,

0 commit comments

Comments
 (0)