Antonio Sanchez 17268b155d Add bit_cast for half/bfloat to/from uint16_t, fix TensorRandom ...

The existing `TensorRandom.h` implementation makes the assumption that
`half` (`bfloat16`) has a `uint16_t` member `x` (`value`), which is not
always true. This currently fails on arm64, where `x` has type `__fp16`.
Added `bit_cast` specializations to allow casting to/from `uint16_t`
for both `half` and `bfloat16`.  Also added tests in
`half_float`, `bfloat16_float`, and `cxx11_tensor_random` to catch
these errors in the future.

2020-11-18 20:32:35 +00:00

20 KiB

Raw Blame History

View Raw