Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
[ghstack-poisoned]
swolchok committed Jan 27, 2025
1 parent da0d85d commit a684e30
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 a684e30

Please sign in to comment.