bfloat16_float.cpp 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378
  1. // This file is part of Eigen, a lightweight C++ template library
  2. // for linear algebra.
  3. //
  4. // This Source Code Form is subject to the terms of the Mozilla
  5. // Public License v. 2.0. If a copy of the MPL was not distributed
  6. // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
  7. #include <sstream>
  8. #include <memory>
  9. #include <math.h>
  10. #include "main.h"
  11. #include <Eigen/src/Core/arch/Default/BFloat16.h>
  12. #define VERIFY_BFLOAT16_BITS_EQUAL(h, bits) \
  13. VERIFY_IS_EQUAL((numext::bit_cast<numext::uint16_t>(h)), (static_cast<numext::uint16_t>(bits)))
  14. // Make sure it's possible to forward declare Eigen::bfloat16
  15. namespace Eigen {
  16. struct bfloat16;
  17. }
  18. using Eigen::bfloat16;
  19. float BinaryToFloat(uint32_t sign, uint32_t exponent, uint32_t high_mantissa,
  20. uint32_t low_mantissa) {
  21. float dest;
  22. uint32_t src = (sign << 31) + (exponent << 23) + (high_mantissa << 16) + low_mantissa;
  23. memcpy(static_cast<void*>(&dest),
  24. static_cast<const void*>(&src), sizeof(dest));
  25. return dest;
  26. }
  27. template<typename T>
  28. void test_roundtrip() {
  29. // Representable T round trip via bfloat16
  30. VERIFY_IS_EQUAL((internal::cast<bfloat16,T>(internal::cast<T,bfloat16>(-std::numeric_limits<T>::infinity()))), -std::numeric_limits<T>::infinity());
  31. VERIFY_IS_EQUAL((internal::cast<bfloat16,T>(internal::cast<T,bfloat16>(std::numeric_limits<T>::infinity()))), std::numeric_limits<T>::infinity());
  32. VERIFY_IS_EQUAL((internal::cast<bfloat16,T>(internal::cast<T,bfloat16>(T(-1.0)))), T(-1.0));
  33. VERIFY_IS_EQUAL((internal::cast<bfloat16,T>(internal::cast<T,bfloat16>(T(-0.5)))), T(-0.5));
  34. VERIFY_IS_EQUAL((internal::cast<bfloat16,T>(internal::cast<T,bfloat16>(T(-0.0)))), T(-0.0));
  35. VERIFY_IS_EQUAL((internal::cast<bfloat16,T>(internal::cast<T,bfloat16>(T(1.0)))), T(1.0));
  36. VERIFY_IS_EQUAL((internal::cast<bfloat16,T>(internal::cast<T,bfloat16>(T(0.5)))), T(0.5));
  37. VERIFY_IS_EQUAL((internal::cast<bfloat16,T>(internal::cast<T,bfloat16>(T(0.0)))), T(0.0));
  38. }
  39. void test_conversion()
  40. {
  41. using Eigen::bfloat16_impl::__bfloat16_raw;
  42. // Round-trip casts
  43. VERIFY_IS_EQUAL(
  44. numext::bit_cast<bfloat16>(numext::bit_cast<numext::uint16_t>(bfloat16(1.0f))),
  45. bfloat16(1.0f));
  46. VERIFY_IS_EQUAL(
  47. numext::bit_cast<bfloat16>(numext::bit_cast<numext::uint16_t>(bfloat16(0.5f))),
  48. bfloat16(0.5f));
  49. VERIFY_IS_EQUAL(
  50. numext::bit_cast<bfloat16>(numext::bit_cast<numext::uint16_t>(bfloat16(-0.33333f))),
  51. bfloat16(-0.33333f));
  52. VERIFY_IS_EQUAL(
  53. numext::bit_cast<bfloat16>(numext::bit_cast<numext::uint16_t>(bfloat16(0.0f))),
  54. bfloat16(0.0f));
  55. // Conversion from float.
  56. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(1.0f), 0x3f80);
  57. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(0.5f), 0x3f00);
  58. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(0.33333f), 0x3eab);
  59. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(3.38e38f), 0x7f7e);
  60. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(3.40e38f), 0x7f80); // Becomes infinity.
  61. // Verify round-to-nearest-even behavior.
  62. float val1 = static_cast<float>(bfloat16(__bfloat16_raw(0x3c00)));
  63. float val2 = static_cast<float>(bfloat16(__bfloat16_raw(0x3c01)));
  64. float val3 = static_cast<float>(bfloat16(__bfloat16_raw(0x3c02)));
  65. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(0.5f * (val1 + val2)), 0x3c00);
  66. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(0.5f * (val2 + val3)), 0x3c02);
  67. // Conversion from int.
  68. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(-1), 0xbf80);
  69. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(0), 0x0000);
  70. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(1), 0x3f80);
  71. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(2), 0x4000);
  72. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(3), 0x4040);
  73. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(12), 0x4140);
  74. // Conversion from bool.
  75. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(false), 0x0000);
  76. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(true), 0x3f80);
  77. // Conversion to bool
  78. VERIFY_IS_EQUAL(static_cast<bool>(bfloat16(3)), true);
  79. VERIFY_IS_EQUAL(static_cast<bool>(bfloat16(0.33333f)), true);
  80. VERIFY_IS_EQUAL(bfloat16(-0.0), false);
  81. VERIFY_IS_EQUAL(static_cast<bool>(bfloat16(0.0)), false);
  82. // Explicit conversion to float.
  83. VERIFY_IS_EQUAL(static_cast<float>(bfloat16(__bfloat16_raw(0x0000))), 0.0f);
  84. VERIFY_IS_EQUAL(static_cast<float>(bfloat16(__bfloat16_raw(0x3f80))), 1.0f);
  85. // Implicit conversion to float
  86. VERIFY_IS_EQUAL(bfloat16(__bfloat16_raw(0x0000)), 0.0f);
  87. VERIFY_IS_EQUAL(bfloat16(__bfloat16_raw(0x3f80)), 1.0f);
  88. // Zero representations
  89. VERIFY_IS_EQUAL(bfloat16(0.0f), bfloat16(0.0f));
  90. VERIFY_IS_EQUAL(bfloat16(-0.0f), bfloat16(0.0f));
  91. VERIFY_IS_EQUAL(bfloat16(-0.0f), bfloat16(-0.0f));
  92. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(0.0f), 0x0000);
  93. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(-0.0f), 0x8000);
  94. // Default is zero
  95. VERIFY_IS_EQUAL(static_cast<float>(bfloat16()), 0.0f);
  96. // Representable floats round trip via bfloat16
  97. test_roundtrip<float>();
  98. test_roundtrip<double>();
  99. test_roundtrip<std::complex<float> >();
  100. test_roundtrip<std::complex<double> >();
  101. // Conversion
  102. Array<float,1,100> a;
  103. for (int i = 0; i < 100; i++) a(i) = i + 1.25;
  104. Array<bfloat16,1,100> b = a.cast<bfloat16>();
  105. Array<float,1,100> c = b.cast<float>();
  106. for (int i = 0; i < 100; ++i) {
  107. VERIFY_LE(numext::abs(c(i) - a(i)), a(i) / 128);
  108. }
  109. // Epsilon
  110. VERIFY_LE(1.0f, static_cast<float>((std::numeric_limits<bfloat16>::epsilon)() + bfloat16(1.0f)));
  111. VERIFY_IS_EQUAL(1.0f, static_cast<float>((std::numeric_limits<bfloat16>::epsilon)() / bfloat16(2.0f) + bfloat16(1.0f)));
  112. // Negate
  113. VERIFY_IS_EQUAL(static_cast<float>(-bfloat16(3.0f)), -3.0f);
  114. VERIFY_IS_EQUAL(static_cast<float>(-bfloat16(-4.5f)), 4.5f);
  115. #if !EIGEN_COMP_MSVC
  116. // Visual Studio errors out on divisions by 0
  117. VERIFY((numext::isnan)(static_cast<float>(bfloat16(0.0 / 0.0))));
  118. VERIFY((numext::isinf)(static_cast<float>(bfloat16(1.0 / 0.0))));
  119. VERIFY((numext::isinf)(static_cast<float>(bfloat16(-1.0 / 0.0))));
  120. // Visual Studio errors out on divisions by 0
  121. VERIFY((numext::isnan)(bfloat16(0.0 / 0.0)));
  122. VERIFY((numext::isinf)(bfloat16(1.0 / 0.0)));
  123. VERIFY((numext::isinf)(bfloat16(-1.0 / 0.0)));
  124. #endif
  125. // NaNs and infinities.
  126. VERIFY(!(numext::isinf)(static_cast<float>(bfloat16(3.38e38f)))); // Largest finite number.
  127. VERIFY(!(numext::isnan)(static_cast<float>(bfloat16(0.0f))));
  128. VERIFY((numext::isinf)(static_cast<float>(bfloat16(__bfloat16_raw(0xff80)))));
  129. VERIFY((numext::isnan)(static_cast<float>(bfloat16(__bfloat16_raw(0xffc0)))));
  130. VERIFY((numext::isinf)(static_cast<float>(bfloat16(__bfloat16_raw(0x7f80)))));
  131. VERIFY((numext::isnan)(static_cast<float>(bfloat16(__bfloat16_raw(0x7fc0)))));
  132. // Exactly same checks as above, just directly on the bfloat16 representation.
  133. VERIFY(!(numext::isinf)(bfloat16(__bfloat16_raw(0x7bff))));
  134. VERIFY(!(numext::isnan)(bfloat16(__bfloat16_raw(0x0000))));
  135. VERIFY((numext::isinf)(bfloat16(__bfloat16_raw(0xff80))));
  136. VERIFY((numext::isnan)(bfloat16(__bfloat16_raw(0xffc0))));
  137. VERIFY((numext::isinf)(bfloat16(__bfloat16_raw(0x7f80))));
  138. VERIFY((numext::isnan)(bfloat16(__bfloat16_raw(0x7fc0))));
  139. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(BinaryToFloat(0x0, 0xff, 0x40, 0x0)), 0x7fc0);
  140. VERIFY_BFLOAT16_BITS_EQUAL(bfloat16(BinaryToFloat(0x1, 0xff, 0x40, 0x0)), 0xffc0);
  141. }
  142. void test_numtraits()
  143. {
  144. std::cout << "epsilon = " << NumTraits<bfloat16>::epsilon() << " (0x" << std::hex << numext::bit_cast<numext::uint16_t>(NumTraits<bfloat16>::epsilon()) << ")" << std::endl;
  145. std::cout << "highest = " << NumTraits<bfloat16>::highest() << " (0x" << std::hex << numext::bit_cast<numext::uint16_t>(NumTraits<bfloat16>::highest()) << ")" << std::endl;
  146. std::cout << "lowest = " << NumTraits<bfloat16>::lowest() << " (0x" << std::hex << numext::bit_cast<numext::uint16_t>(NumTraits<bfloat16>::lowest()) << ")" << std::endl;
  147. std::cout << "min = " << (std::numeric_limits<bfloat16>::min)() << " (0x" << std::hex << numext::bit_cast<numext::uint16_t>((std::numeric_limits<bfloat16>::min)()) << ")" << std::endl;
  148. std::cout << "denorm min = " << (std::numeric_limits<bfloat16>::denorm_min)() << " (0x" << std::hex << numext::bit_cast<numext::uint16_t>((std::numeric_limits<bfloat16>::denorm_min)()) << ")" << std::endl;
  149. std::cout << "infinity = " << NumTraits<bfloat16>::infinity() << " (0x" << std::hex << numext::bit_cast<numext::uint16_t>(NumTraits<bfloat16>::infinity()) << ")" << std::endl;
  150. std::cout << "quiet nan = " << NumTraits<bfloat16>::quiet_NaN() << " (0x" << std::hex << numext::bit_cast<numext::uint16_t>(NumTraits<bfloat16>::quiet_NaN()) << ")" << std::endl;
  151. std::cout << "signaling nan = " << std::numeric_limits<bfloat16>::signaling_NaN() << " (0x" << std::hex << numext::bit_cast<numext::uint16_t>(std::numeric_limits<bfloat16>::signaling_NaN()) << ")" << std::endl;
  152. VERIFY(NumTraits<bfloat16>::IsSigned);
  153. VERIFY_IS_EQUAL(
  154. numext::bit_cast<numext::uint16_t>(std::numeric_limits<bfloat16>::infinity()),
  155. numext::bit_cast<numext::uint16_t>(bfloat16(std::numeric_limits<float>::infinity())) );
  156. // There is no guarantee that casting a 32-bit NaN to bfloat16 has a precise
  157. // bit pattern. We test that it is in fact a NaN, then test the signaling
  158. // bit (msb of significand is 1 for quiet, 0 for signaling).
  159. const numext::uint16_t BFLOAT16_QUIET_BIT = 0x0040;
  160. VERIFY(
  161. (numext::isnan)(std::numeric_limits<bfloat16>::quiet_NaN())
  162. && (numext::isnan)(bfloat16(std::numeric_limits<float>::quiet_NaN()))
  163. && ((numext::bit_cast<numext::uint16_t>(std::numeric_limits<bfloat16>::quiet_NaN()) & BFLOAT16_QUIET_BIT) > 0)
  164. && ((numext::bit_cast<numext::uint16_t>(bfloat16(std::numeric_limits<float>::quiet_NaN())) & BFLOAT16_QUIET_BIT) > 0) );
  165. // After a cast to bfloat16, a signaling NaN may become non-signaling. Thus,
  166. // we check that both are NaN, and that only the `numeric_limits` version is
  167. // signaling.
  168. VERIFY(
  169. (numext::isnan)(std::numeric_limits<bfloat16>::signaling_NaN())
  170. && (numext::isnan)(bfloat16(std::numeric_limits<float>::signaling_NaN()))
  171. && ((numext::bit_cast<numext::uint16_t>(std::numeric_limits<bfloat16>::signaling_NaN()) & BFLOAT16_QUIET_BIT) == 0) );
  172. VERIFY( (std::numeric_limits<bfloat16>::min)() > bfloat16(0.f) );
  173. VERIFY( (std::numeric_limits<bfloat16>::denorm_min)() > bfloat16(0.f) );
  174. VERIFY_IS_EQUAL( (std::numeric_limits<bfloat16>::denorm_min)()/bfloat16(2), bfloat16(0.f) );
  175. }
  176. void test_arithmetic()
  177. {
  178. VERIFY_IS_EQUAL(static_cast<float>(bfloat16(2) + bfloat16(2)), 4);
  179. VERIFY_IS_EQUAL(static_cast<float>(bfloat16(2) + bfloat16(-2)), 0);
  180. VERIFY_IS_APPROX(static_cast<float>(bfloat16(0.33333f) + bfloat16(0.66667f)), 1.0f);
  181. VERIFY_IS_EQUAL(static_cast<float>(bfloat16(2.0f) * bfloat16(-5.5f)), -11.0f);
  182. VERIFY_IS_APPROX(static_cast<float>(bfloat16(1.0f) / bfloat16(3.0f)), 0.3339f);
  183. VERIFY_IS_EQUAL(static_cast<float>(-bfloat16(4096.0f)), -4096.0f);
  184. VERIFY_IS_EQUAL(static_cast<float>(-bfloat16(-4096.0f)), 4096.0f);
  185. }
  186. void test_comparison()
  187. {
  188. VERIFY(bfloat16(1.0f) > bfloat16(0.5f));
  189. VERIFY(bfloat16(0.5f) < bfloat16(1.0f));
  190. VERIFY(!(bfloat16(1.0f) < bfloat16(0.5f)));
  191. VERIFY(!(bfloat16(0.5f) > bfloat16(1.0f)));
  192. VERIFY(!(bfloat16(4.0f) > bfloat16(4.0f)));
  193. VERIFY(!(bfloat16(4.0f) < bfloat16(4.0f)));
  194. VERIFY(!(bfloat16(0.0f) < bfloat16(-0.0f)));
  195. VERIFY(!(bfloat16(-0.0f) < bfloat16(0.0f)));
  196. VERIFY(!(bfloat16(0.0f) > bfloat16(-0.0f)));
  197. VERIFY(!(bfloat16(-0.0f) > bfloat16(0.0f)));
  198. VERIFY(bfloat16(0.2f) > bfloat16(-1.0f));
  199. VERIFY(bfloat16(-1.0f) < bfloat16(0.2f));
  200. VERIFY(bfloat16(-16.0f) < bfloat16(-15.0f));
  201. VERIFY(bfloat16(1.0f) == bfloat16(1.0f));
  202. VERIFY(bfloat16(1.0f) != bfloat16(2.0f));
  203. // Comparisons with NaNs and infinities.
  204. #if !EIGEN_COMP_MSVC
  205. // Visual Studio errors out on divisions by 0
  206. VERIFY(!(bfloat16(0.0 / 0.0) == bfloat16(0.0 / 0.0)));
  207. VERIFY(bfloat16(0.0 / 0.0) != bfloat16(0.0 / 0.0));
  208. VERIFY(!(bfloat16(1.0) == bfloat16(0.0 / 0.0)));
  209. VERIFY(!(bfloat16(1.0) < bfloat16(0.0 / 0.0)));
  210. VERIFY(!(bfloat16(1.0) > bfloat16(0.0 / 0.0)));
  211. VERIFY(bfloat16(1.0) != bfloat16(0.0 / 0.0));
  212. VERIFY(bfloat16(1.0) < bfloat16(1.0 / 0.0));
  213. VERIFY(bfloat16(1.0) > bfloat16(-1.0 / 0.0));
  214. #endif
  215. }
  216. void test_basic_functions()
  217. {
  218. VERIFY_IS_EQUAL(static_cast<float>(numext::abs(bfloat16(3.5f))), 3.5f);
  219. VERIFY_IS_EQUAL(static_cast<float>(abs(bfloat16(3.5f))), 3.5f);
  220. VERIFY_IS_EQUAL(static_cast<float>(numext::abs(bfloat16(-3.5f))), 3.5f);
  221. VERIFY_IS_EQUAL(static_cast<float>(abs(bfloat16(-3.5f))), 3.5f);
  222. VERIFY_IS_EQUAL(static_cast<float>(numext::floor(bfloat16(3.5f))), 3.0f);
  223. VERIFY_IS_EQUAL(static_cast<float>(floor(bfloat16(3.5f))), 3.0f);
  224. VERIFY_IS_EQUAL(static_cast<float>(numext::floor(bfloat16(-3.5f))), -4.0f);
  225. VERIFY_IS_EQUAL(static_cast<float>(floor(bfloat16(-3.5f))), -4.0f);
  226. VERIFY_IS_EQUAL(static_cast<float>(numext::ceil(bfloat16(3.5f))), 4.0f);
  227. VERIFY_IS_EQUAL(static_cast<float>(ceil(bfloat16(3.5f))), 4.0f);
  228. VERIFY_IS_EQUAL(static_cast<float>(numext::ceil(bfloat16(-3.5f))), -3.0f);
  229. VERIFY_IS_EQUAL(static_cast<float>(ceil(bfloat16(-3.5f))), -3.0f);
  230. VERIFY_IS_APPROX(static_cast<float>(numext::sqrt(bfloat16(0.0f))), 0.0f);
  231. VERIFY_IS_APPROX(static_cast<float>(sqrt(bfloat16(0.0f))), 0.0f);
  232. VERIFY_IS_APPROX(static_cast<float>(numext::sqrt(bfloat16(4.0f))), 2.0f);
  233. VERIFY_IS_APPROX(static_cast<float>(sqrt(bfloat16(4.0f))), 2.0f);
  234. VERIFY_IS_APPROX(static_cast<float>(numext::pow(bfloat16(0.0f), bfloat16(1.0f))), 0.0f);
  235. VERIFY_IS_APPROX(static_cast<float>(pow(bfloat16(0.0f), bfloat16(1.0f))), 0.0f);
  236. VERIFY_IS_APPROX(static_cast<float>(numext::pow(bfloat16(2.0f), bfloat16(2.0f))), 4.0f);
  237. VERIFY_IS_APPROX(static_cast<float>(pow(bfloat16(2.0f), bfloat16(2.0f))), 4.0f);
  238. VERIFY_IS_EQUAL(static_cast<float>(numext::exp(bfloat16(0.0f))), 1.0f);
  239. VERIFY_IS_EQUAL(static_cast<float>(exp(bfloat16(0.0f))), 1.0f);
  240. VERIFY_IS_APPROX(static_cast<float>(numext::exp(bfloat16(EIGEN_PI))), 20.f + static_cast<float>(EIGEN_PI));
  241. VERIFY_IS_APPROX(static_cast<float>(exp(bfloat16(EIGEN_PI))), 20.f + static_cast<float>(EIGEN_PI));
  242. VERIFY_IS_EQUAL(static_cast<float>(numext::expm1(bfloat16(0.0f))), 0.0f);
  243. VERIFY_IS_EQUAL(static_cast<float>(expm1(bfloat16(0.0f))), 0.0f);
  244. VERIFY_IS_APPROX(static_cast<float>(numext::expm1(bfloat16(2.0f))), 6.375f);
  245. VERIFY_IS_APPROX(static_cast<float>(expm1(bfloat16(2.0f))), 6.375f);
  246. VERIFY_IS_EQUAL(static_cast<float>(numext::log(bfloat16(1.0f))), 0.0f);
  247. VERIFY_IS_EQUAL(static_cast<float>(log(bfloat16(1.0f))), 0.0f);
  248. VERIFY_IS_APPROX(static_cast<float>(numext::log(bfloat16(10.0f))), 2.296875f);
  249. VERIFY_IS_APPROX(static_cast<float>(log(bfloat16(10.0f))), 2.296875f);
  250. VERIFY_IS_EQUAL(static_cast<float>(numext::log1p(bfloat16(0.0f))), 0.0f);
  251. VERIFY_IS_EQUAL(static_cast<float>(log1p(bfloat16(0.0f))), 0.0f);
  252. VERIFY_IS_APPROX(static_cast<float>(numext::log1p(bfloat16(10.0f))), 2.390625f);
  253. VERIFY_IS_APPROX(static_cast<float>(log1p(bfloat16(10.0f))), 2.390625f);
  254. }
  255. void test_trigonometric_functions()
  256. {
  257. VERIFY_IS_APPROX(numext::cos(bfloat16(0.0f)), bfloat16(cosf(0.0f)));
  258. VERIFY_IS_APPROX(cos(bfloat16(0.0f)), bfloat16(cosf(0.0f)));
  259. VERIFY_IS_APPROX(numext::cos(bfloat16(EIGEN_PI)), bfloat16(cosf(EIGEN_PI)));
  260. // VERIFY_IS_APPROX(numext::cos(bfloat16(EIGEN_PI/2)), bfloat16(cosf(EIGEN_PI/2)));
  261. // VERIFY_IS_APPROX(numext::cos(bfloat16(3*EIGEN_PI/2)), bfloat16(cosf(3*EIGEN_PI/2)));
  262. VERIFY_IS_APPROX(numext::cos(bfloat16(3.5f)), bfloat16(cosf(3.5f)));
  263. VERIFY_IS_APPROX(numext::sin(bfloat16(0.0f)), bfloat16(sinf(0.0f)));
  264. VERIFY_IS_APPROX(sin(bfloat16(0.0f)), bfloat16(sinf(0.0f)));
  265. // VERIFY_IS_APPROX(numext::sin(bfloat16(EIGEN_PI)), bfloat16(sinf(EIGEN_PI)));
  266. VERIFY_IS_APPROX(numext::sin(bfloat16(EIGEN_PI/2)), bfloat16(sinf(EIGEN_PI/2)));
  267. VERIFY_IS_APPROX(numext::sin(bfloat16(3*EIGEN_PI/2)), bfloat16(sinf(3*EIGEN_PI/2)));
  268. VERIFY_IS_APPROX(numext::sin(bfloat16(3.5f)), bfloat16(sinf(3.5f)));
  269. VERIFY_IS_APPROX(numext::tan(bfloat16(0.0f)), bfloat16(tanf(0.0f)));
  270. VERIFY_IS_APPROX(tan(bfloat16(0.0f)), bfloat16(tanf(0.0f)));
  271. // VERIFY_IS_APPROX(numext::tan(bfloat16(EIGEN_PI)), bfloat16(tanf(EIGEN_PI)));
  272. // VERIFY_IS_APPROX(numext::tan(bfloat16(EIGEN_PI/2)), bfloat16(tanf(EIGEN_PI/2)));
  273. // VERIFY_IS_APPROX(numext::tan(bfloat16(3*EIGEN_PI/2)), bfloat16(tanf(3*EIGEN_PI/2)));
  274. VERIFY_IS_APPROX(numext::tan(bfloat16(3.5f)), bfloat16(tanf(3.5f)));
  275. }
  276. void test_array()
  277. {
  278. typedef Array<bfloat16,1,Dynamic> ArrayXh;
  279. Index size = internal::random<Index>(1,10);
  280. Index i = internal::random<Index>(0,size-1);
  281. ArrayXh a1 = ArrayXh::Random(size), a2 = ArrayXh::Random(size);
  282. VERIFY_IS_APPROX( a1+a1, bfloat16(2)*a1 );
  283. VERIFY( (a1.abs() >= bfloat16(0)).all() );
  284. VERIFY_IS_APPROX( (a1*a1).sqrt(), a1.abs() );
  285. VERIFY( ((a1.min)(a2) <= (a1.max)(a2)).all() );
  286. a1(i) = bfloat16(-10.);
  287. VERIFY_IS_EQUAL( a1.minCoeff(), bfloat16(-10.) );
  288. a1(i) = bfloat16(10.);
  289. VERIFY_IS_EQUAL( a1.maxCoeff(), bfloat16(10.) );
  290. std::stringstream ss;
  291. ss << a1;
  292. }
  293. void test_product()
  294. {
  295. typedef Matrix<bfloat16,Dynamic,Dynamic> MatrixXh;
  296. Index rows = internal::random<Index>(1,EIGEN_TEST_MAX_SIZE);
  297. Index cols = internal::random<Index>(1,EIGEN_TEST_MAX_SIZE);
  298. Index depth = internal::random<Index>(1,EIGEN_TEST_MAX_SIZE);
  299. MatrixXh Ah = MatrixXh::Random(rows,depth);
  300. MatrixXh Bh = MatrixXh::Random(depth,cols);
  301. MatrixXh Ch = MatrixXh::Random(rows,cols);
  302. MatrixXf Af = Ah.cast<float>();
  303. MatrixXf Bf = Bh.cast<float>();
  304. MatrixXf Cf = Ch.cast<float>();
  305. VERIFY_IS_APPROX(Ch.noalias()+=Ah*Bh, (Cf.noalias()+=Af*Bf).cast<bfloat16>());
  306. }
  307. EIGEN_DECLARE_TEST(bfloat16_float)
  308. {
  309. CALL_SUBTEST(test_numtraits());
  310. for(int i = 0; i < g_repeat; i++) {
  311. CALL_SUBTEST(test_conversion());
  312. CALL_SUBTEST(test_arithmetic());
  313. CALL_SUBTEST(test_comparison());
  314. CALL_SUBTEST(test_basic_functions());
  315. CALL_SUBTEST(test_trigonometric_functions());
  316. CALL_SUBTEST(test_array());
  317. CALL_SUBTEST(test_product());
  318. }
  319. }