Unbreak custom_kernel_example test

This test doesn't run in OSS CI (#7984), so I missed it. Test Plan: ran relevant test internally with buck2 ghstack-source-id: f2b994d3bf1091939bad51f5f8455085e91ed8af ghstack-comment-id: 2617137527 Pull Request resolved: #7985
pytorch · Jan 27, 2025 · 01d4186 · 01d4186
1 parent da0d85d
commit 01d4186
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,