From a684e3074237e9582776d6e30eb7ce8121b1f59d Mon Sep 17 00:00:00 2001 From: Scott Wolchok Date: Mon, 27 Jan 2025 15:37:34 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- kernels/test/custom_kernel_example/op_relu.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kernels/test/custom_kernel_example/op_relu.cpp b/kernels/test/custom_kernel_example/op_relu.cpp index 3ea487dffa..b5faa7a013 100644 --- a/kernels/test/custom_kernel_example/op_relu.cpp +++ b/kernels/test/custom_kernel_example/op_relu.cpp @@ -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,