diff options
author | David Tellenbach <david.tellenbach@me.com> | 2020-10-29 07:28:28 +0100 |
---|---|---|
committer | David Tellenbach <david.tellenbach@me.com> | 2020-10-29 07:28:28 +0100 |
commit | 09f015852bac3c28849aa9bbcf0e3cc1c2fb6808 (patch) | |
tree | c1bc9c99ec60db40cc6f3052af57605af33a9b51 /Eigen/src/Core/arch/Default/BFloat16.h | |
parent | e265f7ed8e59c26e15f2c35162c6b8da1c5d594f (diff) |
Replace numext::as_uint with numext::bit_cast<numext::uint32_t>
Diffstat (limited to 'Eigen/src/Core/arch/Default/BFloat16.h')
-rw-r--r-- | Eigen/src/Core/arch/Default/BFloat16.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/Eigen/src/Core/arch/Default/BFloat16.h b/Eigen/src/Core/arch/Default/BFloat16.h index 95a84e488..63ceace1e 100644 --- a/Eigen/src/Core/arch/Default/BFloat16.h +++ b/Eigen/src/Core/arch/Default/BFloat16.h @@ -460,14 +460,14 @@ EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC __bfloat16_raw float_to_bfloat16_rtne<true #if (defined(EIGEN_HAS_CUDA_BF16) && defined(EIGEN_HAS_HIP_BF16)) // Nothing to do here #else - unsigned int input = numext::as_uint(ff); + numext::uint32_t input = numext::bit_cast<numext::uint32_t>(ff); __bfloat16_raw output; // Least significant bit of resulting bfloat. - unsigned int lsb = (input >> 16) & 1; - unsigned int rounding_bias = 0x7fff + lsb; + numext::uint32_t lsb = (input >> 16) & 1; + numext::uint32_t rounding_bias = 0x7fff + lsb; input += rounding_bias; - output.value = static_cast<unsigned short>(input >> 16); + output.value = static_cast<numext::uint16_t>(input >> 16); return output; #endif } |