Check both flags to preserve NaN

e803354a · Rosty Geyyer · 15eedb2b · e803354a
Commit e803354a authored Mar 15, 2023 by Rosty Geyyer
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/utility/data_type.hpp include/ck/utility/data_type.hpp +1 -1

No files found.
--- a/include/ck/utility/data_type.hpp
+++ b/include/ck/utility/data_type.hpp
@@ -1003,7 +1003,7 @@ inline __host__ __device__ constexpr bhalf_t type_convert<bhalf_t, float>(float
    bool flag1 = u.int32 & 0xffff;

    u.int32 += flag0 ? 0x7fff + ((u.int32 >> 16) & 1) : 0; // Round to nearest, round to even
-    u.int32 |= flag1 ? 0x10000 : 0x0;                      // Preserve signaling NaN
+    u.int32 |= (!flag0 && flag1) ? 0x10000 : 0x0;                      // Preserve signaling NaN

    return uint16_t(u.int32 >> 16);
 }