Skip to content

Commit f4d1926

Browse files
committed
Fix incorrect implementation of not_equal_to for 16-bit floats
1 parent 3f622bc commit f4d1926

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

include/kernel_float/bf16.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ KERNEL_FLOAT_BF16_BINARY_FUN(max, __hmax, __hmax2)
117117
KERNEL_FLOAT_BF16_BINARY_FUN(fast_div, __hdiv, __h2div)
118118

119119
KERNEL_FLOAT_BF16_BINARY_FUN(equal_to, __heq, __heq2)
120-
KERNEL_FLOAT_BF16_BINARY_FUN(not_equal_to, __heq, __heq2)
120+
KERNEL_FLOAT_BF16_BINARY_FUN(not_equal_to, __hneu, __hneu2)
121121
KERNEL_FLOAT_BF16_BINARY_FUN(less, __hlt, __hlt2)
122122
KERNEL_FLOAT_BF16_BINARY_FUN(less_equal, __hle, __hle2)
123123
KERNEL_FLOAT_BF16_BINARY_FUN(greater, __hgt, __hgt2)

include/kernel_float/fp16.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ KERNEL_FLOAT_FP16_BINARY_FUN(max, __hmax, __hmax2)
107107
KERNEL_FLOAT_FP16_BINARY_FUN(fast_div, __hdiv, __h2div)
108108

109109
KERNEL_FLOAT_FP16_BINARY_FUN(equal_to, __heq, __heq2)
110-
KERNEL_FLOAT_FP16_BINARY_FUN(not_equal_to, __heq, __heq2)
110+
KERNEL_FLOAT_FP16_BINARY_FUN(not_equal_to, __hneu, __hneu2)
111111
KERNEL_FLOAT_FP16_BINARY_FUN(less, __hlt, __hlt2)
112112
KERNEL_FLOAT_FP16_BINARY_FUN(less_equal, __hle, __hle2)
113113
KERNEL_FLOAT_FP16_BINARY_FUN(greater, __hgt, __hgt2)

0 commit comments

Comments
 (0)