packetmath.cpp 54 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302
  1. // This file is part of Eigen, a lightweight C++ template library
  2. // for linear algebra.
  3. //
  4. // Copyright (C) 2008-2009 Gael Guennebaud <gael.guennebaud@inria.fr>
  5. // Copyright (C) 2006-2008 Benoit Jacob <jacob.benoit.1@gmail.com>
  6. //
  7. // This Source Code Form is subject to the terms of the Mozilla
  8. // Public License v. 2.0. If a copy of the MPL was not distributed
  9. // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
  10. #include "packetmath_test_shared.h"
  11. #include "random_without_cast_overflow.h"
  12. template <typename T>
  13. inline T REF_ADD(const T& a, const T& b) {
  14. return a + b;
  15. }
  16. template <typename T>
  17. inline T REF_SUB(const T& a, const T& b) {
  18. return a - b;
  19. }
  20. template <typename T>
  21. inline T REF_MUL(const T& a, const T& b) {
  22. return a * b;
  23. }
  24. template <typename T>
  25. inline T REF_DIV(const T& a, const T& b) {
  26. return a / b;
  27. }
  28. template <typename T>
  29. inline T REF_ABS_DIFF(const T& a, const T& b) {
  30. return a > b ? a - b : b - a;
  31. }
  32. // Specializations for bool.
  33. template <>
  34. inline bool REF_ADD(const bool& a, const bool& b) {
  35. return a || b;
  36. }
  37. template <>
  38. inline bool REF_SUB(const bool& a, const bool& b) {
  39. return a ^ b;
  40. }
  41. template <>
  42. inline bool REF_MUL(const bool& a, const bool& b) {
  43. return a && b;
  44. }
  45. template <typename T>
  46. inline T REF_FREXP(const T& x, T& exp) {
  47. int iexp;
  48. EIGEN_USING_STD(frexp)
  49. const T out = static_cast<T>(frexp(x, &iexp));
  50. exp = static_cast<T>(iexp);
  51. return out;
  52. }
  53. template <typename T>
  54. inline T REF_LDEXP(const T& x, const T& exp) {
  55. EIGEN_USING_STD(ldexp)
  56. return static_cast<T>(ldexp(x, static_cast<int>(exp)));
  57. }
  58. // Uses pcast to cast from one array to another.
  59. template <typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int TgtCoeffRatio>
  60. struct pcast_array;
  61. template <typename SrcPacket, typename TgtPacket, int TgtCoeffRatio>
  62. struct pcast_array<SrcPacket, TgtPacket, 1, TgtCoeffRatio> {
  63. typedef typename internal::unpacket_traits<SrcPacket>::type SrcScalar;
  64. typedef typename internal::unpacket_traits<TgtPacket>::type TgtScalar;
  65. static void cast(const SrcScalar* src, size_t size, TgtScalar* dst) {
  66. static const int SrcPacketSize = internal::unpacket_traits<SrcPacket>::size;
  67. static const int TgtPacketSize = internal::unpacket_traits<TgtPacket>::size;
  68. size_t i;
  69. for (i = 0; i < size && i + SrcPacketSize <= size; i += TgtPacketSize) {
  70. internal::pstoreu(dst + i, internal::pcast<SrcPacket, TgtPacket>(internal::ploadu<SrcPacket>(src + i)));
  71. }
  72. // Leftovers that cannot be loaded into a packet.
  73. for (; i < size; ++i) {
  74. dst[i] = static_cast<TgtScalar>(src[i]);
  75. }
  76. }
  77. };
  78. template <typename SrcPacket, typename TgtPacket>
  79. struct pcast_array<SrcPacket, TgtPacket, 2, 1> {
  80. static void cast(const typename internal::unpacket_traits<SrcPacket>::type* src, size_t size,
  81. typename internal::unpacket_traits<TgtPacket>::type* dst) {
  82. static const int SrcPacketSize = internal::unpacket_traits<SrcPacket>::size;
  83. static const int TgtPacketSize = internal::unpacket_traits<TgtPacket>::size;
  84. for (size_t i = 0; i < size; i += TgtPacketSize) {
  85. SrcPacket a = internal::ploadu<SrcPacket>(src + i);
  86. SrcPacket b = internal::ploadu<SrcPacket>(src + i + SrcPacketSize);
  87. internal::pstoreu(dst + i, internal::pcast<SrcPacket, TgtPacket>(a, b));
  88. }
  89. }
  90. };
  91. template <typename SrcPacket, typename TgtPacket>
  92. struct pcast_array<SrcPacket, TgtPacket, 4, 1> {
  93. static void cast(const typename internal::unpacket_traits<SrcPacket>::type* src, size_t size,
  94. typename internal::unpacket_traits<TgtPacket>::type* dst) {
  95. static const int SrcPacketSize = internal::unpacket_traits<SrcPacket>::size;
  96. static const int TgtPacketSize = internal::unpacket_traits<TgtPacket>::size;
  97. for (size_t i = 0; i < size; i += TgtPacketSize) {
  98. SrcPacket a = internal::ploadu<SrcPacket>(src + i);
  99. SrcPacket b = internal::ploadu<SrcPacket>(src + i + SrcPacketSize);
  100. SrcPacket c = internal::ploadu<SrcPacket>(src + i + 2 * SrcPacketSize);
  101. SrcPacket d = internal::ploadu<SrcPacket>(src + i + 3 * SrcPacketSize);
  102. internal::pstoreu(dst + i, internal::pcast<SrcPacket, TgtPacket>(a, b, c, d));
  103. }
  104. }
  105. };
  106. template <typename SrcPacket, typename TgtPacket>
  107. struct pcast_array<SrcPacket, TgtPacket, 8, 1> {
  108. static void cast(const typename internal::unpacket_traits<SrcPacket>::type* src, size_t size,
  109. typename internal::unpacket_traits<TgtPacket>::type* dst) {
  110. static const int SrcPacketSize = internal::unpacket_traits<SrcPacket>::size;
  111. static const int TgtPacketSize = internal::unpacket_traits<TgtPacket>::size;
  112. for (size_t i = 0; i < size; i += TgtPacketSize) {
  113. SrcPacket a = internal::ploadu<SrcPacket>(src + i);
  114. SrcPacket b = internal::ploadu<SrcPacket>(src + i + SrcPacketSize);
  115. SrcPacket c = internal::ploadu<SrcPacket>(src + i + 2 * SrcPacketSize);
  116. SrcPacket d = internal::ploadu<SrcPacket>(src + i + 3 * SrcPacketSize);
  117. SrcPacket e = internal::ploadu<SrcPacket>(src + i + 4 * SrcPacketSize);
  118. SrcPacket f = internal::ploadu<SrcPacket>(src + i + 5 * SrcPacketSize);
  119. SrcPacket g = internal::ploadu<SrcPacket>(src + i + 6 * SrcPacketSize);
  120. SrcPacket h = internal::ploadu<SrcPacket>(src + i + 7 * SrcPacketSize);
  121. internal::pstoreu(dst + i, internal::pcast<SrcPacket, TgtPacket>(a, b, c, d, e, f, g, h));
  122. }
  123. }
  124. };
  125. template <typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int TgtCoeffRatio, bool CanCast = false>
  126. struct test_cast_helper;
  127. template <typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int TgtCoeffRatio>
  128. struct test_cast_helper<SrcPacket, TgtPacket, SrcCoeffRatio, TgtCoeffRatio, false> {
  129. static void run() {}
  130. };
  131. template <typename SrcPacket, typename TgtPacket, int SrcCoeffRatio, int TgtCoeffRatio>
  132. struct test_cast_helper<SrcPacket, TgtPacket, SrcCoeffRatio, TgtCoeffRatio, true> {
  133. static void run() {
  134. typedef typename internal::unpacket_traits<SrcPacket>::type SrcScalar;
  135. typedef typename internal::unpacket_traits<TgtPacket>::type TgtScalar;
  136. static const int SrcPacketSize = internal::unpacket_traits<SrcPacket>::size;
  137. static const int TgtPacketSize = internal::unpacket_traits<TgtPacket>::size;
  138. static const int BlockSize = SrcPacketSize * SrcCoeffRatio;
  139. eigen_assert(BlockSize == TgtPacketSize * TgtCoeffRatio && "Packet sizes and cast ratios are mismatched.");
  140. static const int DataSize = 10 * BlockSize;
  141. EIGEN_ALIGN_MAX SrcScalar data1[DataSize];
  142. EIGEN_ALIGN_MAX TgtScalar data2[DataSize];
  143. EIGEN_ALIGN_MAX TgtScalar ref[DataSize];
  144. // Construct a packet of scalars that will not overflow when casting
  145. for (int i = 0; i < DataSize; ++i) {
  146. data1[i] = internal::random_without_cast_overflow<SrcScalar, TgtScalar>::value();
  147. }
  148. for (int i = 0; i < DataSize; ++i) {
  149. ref[i] = static_cast<const TgtScalar>(data1[i]);
  150. }
  151. pcast_array<SrcPacket, TgtPacket, SrcCoeffRatio, TgtCoeffRatio>::cast(data1, DataSize, data2);
  152. VERIFY(test::areApprox(ref, data2, DataSize) && "internal::pcast<>");
  153. }
  154. };
  155. template <typename SrcPacket, typename TgtPacket>
  156. struct test_cast {
  157. static void run() {
  158. typedef typename internal::unpacket_traits<SrcPacket>::type SrcScalar;
  159. typedef typename internal::unpacket_traits<TgtPacket>::type TgtScalar;
  160. typedef typename internal::type_casting_traits<SrcScalar, TgtScalar> TypeCastingTraits;
  161. static const int SrcCoeffRatio = TypeCastingTraits::SrcCoeffRatio;
  162. static const int TgtCoeffRatio = TypeCastingTraits::TgtCoeffRatio;
  163. static const int SrcPacketSize = internal::unpacket_traits<SrcPacket>::size;
  164. static const int TgtPacketSize = internal::unpacket_traits<TgtPacket>::size;
  165. static const bool HasCast =
  166. internal::unpacket_traits<SrcPacket>::vectorizable && internal::unpacket_traits<TgtPacket>::vectorizable &&
  167. TypeCastingTraits::VectorizedCast && (SrcPacketSize * SrcCoeffRatio == TgtPacketSize * TgtCoeffRatio);
  168. test_cast_helper<SrcPacket, TgtPacket, SrcCoeffRatio, TgtCoeffRatio, HasCast>::run();
  169. }
  170. };
  171. template <typename SrcPacket, typename TgtScalar,
  172. typename TgtPacket = typename internal::packet_traits<TgtScalar>::type,
  173. bool Vectorized = internal::packet_traits<TgtScalar>::Vectorizable,
  174. bool HasHalf = !internal::is_same<typename internal::unpacket_traits<TgtPacket>::half, TgtPacket>::value>
  175. struct test_cast_runner;
  176. template <typename SrcPacket, typename TgtScalar, typename TgtPacket>
  177. struct test_cast_runner<SrcPacket, TgtScalar, TgtPacket, true, false> {
  178. static void run() { test_cast<SrcPacket, TgtPacket>::run(); }
  179. };
  180. template <typename SrcPacket, typename TgtScalar, typename TgtPacket>
  181. struct test_cast_runner<SrcPacket, TgtScalar, TgtPacket, true, true> {
  182. static void run() {
  183. test_cast<SrcPacket, TgtPacket>::run();
  184. test_cast_runner<SrcPacket, TgtScalar, typename internal::unpacket_traits<TgtPacket>::half>::run();
  185. }
  186. };
  187. template <typename SrcPacket, typename TgtScalar, typename TgtPacket>
  188. struct test_cast_runner<SrcPacket, TgtScalar, TgtPacket, false, false> {
  189. static void run() {}
  190. };
  191. template <typename Scalar, typename Packet, typename EnableIf = void>
  192. struct packetmath_pcast_ops_runner {
  193. static void run() {
  194. test_cast_runner<Packet, float>::run();
  195. test_cast_runner<Packet, double>::run();
  196. test_cast_runner<Packet, int8_t>::run();
  197. test_cast_runner<Packet, uint8_t>::run();
  198. test_cast_runner<Packet, int16_t>::run();
  199. test_cast_runner<Packet, uint16_t>::run();
  200. test_cast_runner<Packet, int32_t>::run();
  201. test_cast_runner<Packet, uint32_t>::run();
  202. test_cast_runner<Packet, int64_t>::run();
  203. test_cast_runner<Packet, uint64_t>::run();
  204. test_cast_runner<Packet, bool>::run();
  205. test_cast_runner<Packet, std::complex<float> >::run();
  206. test_cast_runner<Packet, std::complex<double> >::run();
  207. test_cast_runner<Packet, half>::run();
  208. test_cast_runner<Packet, bfloat16>::run();
  209. }
  210. };
  211. // Only some types support cast from std::complex<>.
  212. template <typename Scalar, typename Packet>
  213. struct packetmath_pcast_ops_runner<Scalar, Packet, typename internal::enable_if<NumTraits<Scalar>::IsComplex>::type> {
  214. static void run() {
  215. test_cast_runner<Packet, std::complex<float> >::run();
  216. test_cast_runner<Packet, std::complex<double> >::run();
  217. test_cast_runner<Packet, half>::run();
  218. test_cast_runner<Packet, bfloat16>::run();
  219. }
  220. };
  221. template <typename Scalar, typename Packet>
  222. void packetmath_boolean_mask_ops() {
  223. const int PacketSize = internal::unpacket_traits<Packet>::size;
  224. const int size = 2 * PacketSize;
  225. EIGEN_ALIGN_MAX Scalar data1[size];
  226. EIGEN_ALIGN_MAX Scalar data2[size];
  227. EIGEN_ALIGN_MAX Scalar ref[size];
  228. for (int i = 0; i < size; ++i) {
  229. data1[i] = internal::random<Scalar>();
  230. }
  231. CHECK_CWISE1(internal::ptrue, internal::ptrue);
  232. CHECK_CWISE2_IF(true, internal::pandnot, internal::pandnot);
  233. for (int i = 0; i < PacketSize; ++i) {
  234. data1[i] = Scalar(i);
  235. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  236. }
  237. CHECK_CWISE2_IF(true, internal::pcmp_eq, internal::pcmp_eq);
  238. //Test (-0) == (0) for signed operations
  239. for (int i = 0; i < PacketSize; ++i) {
  240. data1[i] = Scalar(-0.0);
  241. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  242. }
  243. CHECK_CWISE2_IF(true, internal::pcmp_eq, internal::pcmp_eq);
  244. //Test NaN
  245. for (int i = 0; i < PacketSize; ++i) {
  246. data1[i] = NumTraits<Scalar>::quiet_NaN();
  247. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  248. }
  249. CHECK_CWISE2_IF(true, internal::pcmp_eq, internal::pcmp_eq);
  250. }
  251. template <typename Scalar, typename Packet>
  252. void packetmath_boolean_mask_ops_real() {
  253. const int PacketSize = internal::unpacket_traits<Packet>::size;
  254. const int size = 2 * PacketSize;
  255. EIGEN_ALIGN_MAX Scalar data1[size];
  256. EIGEN_ALIGN_MAX Scalar data2[size];
  257. EIGEN_ALIGN_MAX Scalar ref[size];
  258. for (int i = 0; i < PacketSize; ++i) {
  259. data1[i] = internal::random<Scalar>();
  260. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  261. }
  262. CHECK_CWISE2_IF(true, internal::pcmp_lt_or_nan, internal::pcmp_lt_or_nan);
  263. //Test (-0) <=/< (0) for signed operations
  264. for (int i = 0; i < PacketSize; ++i) {
  265. data1[i] = Scalar(-0.0);
  266. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  267. }
  268. CHECK_CWISE2_IF(true, internal::pcmp_lt_or_nan, internal::pcmp_lt_or_nan);
  269. //Test NaN
  270. for (int i = 0; i < PacketSize; ++i) {
  271. data1[i] = NumTraits<Scalar>::quiet_NaN();
  272. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  273. }
  274. CHECK_CWISE2_IF(true, internal::pcmp_lt_or_nan, internal::pcmp_lt_or_nan);
  275. }
  276. template <typename Scalar, typename Packet>
  277. void packetmath_boolean_mask_ops_notcomplex() {
  278. const int PacketSize = internal::unpacket_traits<Packet>::size;
  279. const int size = 2 * PacketSize;
  280. EIGEN_ALIGN_MAX Scalar data1[size];
  281. EIGEN_ALIGN_MAX Scalar data2[size];
  282. EIGEN_ALIGN_MAX Scalar ref[size];
  283. for (int i = 0; i < PacketSize; ++i) {
  284. data1[i] = internal::random<Scalar>();
  285. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  286. }
  287. CHECK_CWISE2_IF(true, internal::pcmp_le, internal::pcmp_le);
  288. CHECK_CWISE2_IF(true, internal::pcmp_lt, internal::pcmp_lt);
  289. //Test (-0) <=/< (0) for signed operations
  290. for (int i = 0; i < PacketSize; ++i) {
  291. data1[i] = Scalar(-0.0);
  292. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  293. }
  294. CHECK_CWISE2_IF(true, internal::pcmp_le, internal::pcmp_le);
  295. CHECK_CWISE2_IF(true, internal::pcmp_lt, internal::pcmp_lt);
  296. //Test NaN
  297. for (int i = 0; i < PacketSize; ++i) {
  298. data1[i] = NumTraits<Scalar>::quiet_NaN();
  299. data1[i + PacketSize] = internal::random<bool>() ? data1[i] : Scalar(0);
  300. }
  301. CHECK_CWISE2_IF(true, internal::pcmp_le, internal::pcmp_le);
  302. CHECK_CWISE2_IF(true, internal::pcmp_lt, internal::pcmp_lt);
  303. }
  304. // Packet16b representing bool does not support ptrue, pandnot or pcmp_eq, since the scalar path
  305. // (for some compilers) compute the bitwise and with 0x1 of the results to keep the value in [0,1].
  306. template<>
  307. void packetmath_boolean_mask_ops<bool, internal::packet_traits<bool>::type>() {}
  308. template<>
  309. void packetmath_boolean_mask_ops_notcomplex<bool, internal::packet_traits<bool>::type>() {}
  310. template <typename Scalar, typename Packet>
  311. void packetmath_minus_zero_add() {
  312. const int PacketSize = internal::unpacket_traits<Packet>::size;
  313. const int size = 2 * PacketSize;
  314. EIGEN_ALIGN_MAX Scalar data1[size];
  315. EIGEN_ALIGN_MAX Scalar data2[size];
  316. EIGEN_ALIGN_MAX Scalar ref[size];
  317. for (int i = 0; i < PacketSize; ++i) {
  318. data1[i] = Scalar(-0.0);
  319. data1[i + PacketSize] = Scalar(-0.0);
  320. }
  321. CHECK_CWISE2_IF(internal::packet_traits<Scalar>::HasAdd, REF_ADD, internal::padd);
  322. }
  323. // Ensure optimization barrier compiles and doesn't modify contents.
  324. // Only applies to raw types, so will not work for std::complex, Eigen::half
  325. // or Eigen::bfloat16. For those you would need to refer to an underlying
  326. // storage element.
  327. template<typename Packet, typename EnableIf = void>
  328. struct eigen_optimization_barrier_test {
  329. static void run() {}
  330. };
  331. template<typename Packet>
  332. struct eigen_optimization_barrier_test<Packet, typename internal::enable_if<
  333. !NumTraits<Packet>::IsComplex &&
  334. !internal::is_same<Packet, Eigen::half>::value &&
  335. !internal::is_same<Packet, Eigen::bfloat16>::value
  336. >::type> {
  337. static void run() {
  338. typedef typename internal::unpacket_traits<Packet>::type Scalar;
  339. Scalar s = internal::random<Scalar>();
  340. Packet barrier = internal::pset1<Packet>(s);
  341. EIGEN_OPTIMIZATION_BARRIER(barrier);
  342. eigen_assert(s == internal::pfirst(barrier) && "EIGEN_OPTIMIZATION_BARRIER");
  343. }
  344. };
  345. template <typename Scalar, typename Packet>
  346. void packetmath() {
  347. typedef internal::packet_traits<Scalar> PacketTraits;
  348. const int PacketSize = internal::unpacket_traits<Packet>::size;
  349. typedef typename NumTraits<Scalar>::Real RealScalar;
  350. if (g_first_pass)
  351. std::cerr << "=== Testing packet of type '" << typeid(Packet).name() << "' and scalar type '"
  352. << typeid(Scalar).name() << "' and size '" << PacketSize << "' ===\n";
  353. const int max_size = PacketSize > 4 ? PacketSize : 4;
  354. const int size = PacketSize * max_size;
  355. EIGEN_ALIGN_MAX Scalar data1[size];
  356. EIGEN_ALIGN_MAX Scalar data2[size];
  357. EIGEN_ALIGN_MAX Scalar data3[size];
  358. EIGEN_ALIGN_MAX Scalar ref[size];
  359. RealScalar refvalue = RealScalar(0);
  360. eigen_optimization_barrier_test<Packet>::run();
  361. eigen_optimization_barrier_test<Scalar>::run();
  362. for (int i = 0; i < size; ++i) {
  363. data1[i] = internal::random<Scalar>() / RealScalar(PacketSize);
  364. data2[i] = internal::random<Scalar>() / RealScalar(PacketSize);
  365. refvalue = (std::max)(refvalue, numext::abs(data1[i]));
  366. }
  367. internal::pstore(data2, internal::pload<Packet>(data1));
  368. VERIFY(test::areApprox(data1, data2, PacketSize) && "aligned load/store");
  369. for (int offset = 0; offset < PacketSize; ++offset) {
  370. internal::pstore(data2, internal::ploadu<Packet>(data1 + offset));
  371. VERIFY(test::areApprox(data1 + offset, data2, PacketSize) && "internal::ploadu");
  372. }
  373. for (int offset = 0; offset < PacketSize; ++offset) {
  374. internal::pstoreu(data2 + offset, internal::pload<Packet>(data1));
  375. VERIFY(test::areApprox(data1, data2 + offset, PacketSize) && "internal::pstoreu");
  376. }
  377. if (internal::unpacket_traits<Packet>::masked_load_available) {
  378. test::packet_helper<internal::unpacket_traits<Packet>::masked_load_available, Packet> h;
  379. unsigned long long max_umask = (0x1ull << PacketSize);
  380. for (int offset = 0; offset < PacketSize; ++offset) {
  381. for (unsigned long long umask = 0; umask < max_umask; ++umask) {
  382. h.store(data2, h.load(data1 + offset, umask));
  383. for (int k = 0; k < PacketSize; ++k) data3[k] = ((umask & (0x1ull << k)) >> k) ? data1[k + offset] : Scalar(0);
  384. VERIFY(test::areApprox(data3, data2, PacketSize) && "internal::ploadu masked");
  385. }
  386. }
  387. }
  388. if (internal::unpacket_traits<Packet>::masked_store_available) {
  389. test::packet_helper<internal::unpacket_traits<Packet>::masked_store_available, Packet> h;
  390. unsigned long long max_umask = (0x1ull << PacketSize);
  391. for (int offset = 0; offset < PacketSize; ++offset) {
  392. for (unsigned long long umask = 0; umask < max_umask; ++umask) {
  393. internal::pstore(data2, internal::pset1<Packet>(Scalar(0)));
  394. h.store(data2, h.loadu(data1 + offset), umask);
  395. for (int k = 0; k < PacketSize; ++k) data3[k] = ((umask & (0x1ull << k)) >> k) ? data1[k + offset] : Scalar(0);
  396. VERIFY(test::areApprox(data3, data2, PacketSize) && "internal::pstoreu masked");
  397. }
  398. }
  399. }
  400. VERIFY((!PacketTraits::Vectorizable) || PacketTraits::HasAdd);
  401. VERIFY((!PacketTraits::Vectorizable) || PacketTraits::HasSub);
  402. VERIFY((!PacketTraits::Vectorizable) || PacketTraits::HasMul);
  403. CHECK_CWISE2_IF(PacketTraits::HasAdd, REF_ADD, internal::padd);
  404. CHECK_CWISE2_IF(PacketTraits::HasSub, REF_SUB, internal::psub);
  405. CHECK_CWISE2_IF(PacketTraits::HasMul, REF_MUL, internal::pmul);
  406. CHECK_CWISE2_IF(PacketTraits::HasDiv, REF_DIV, internal::pdiv);
  407. if (PacketTraits::HasNegate) CHECK_CWISE1(internal::negate, internal::pnegate);
  408. CHECK_CWISE1(numext::conj, internal::pconj);
  409. for (int offset = 0; offset < 3; ++offset) {
  410. for (int i = 0; i < PacketSize; ++i) ref[i] = data1[offset];
  411. internal::pstore(data2, internal::pset1<Packet>(data1[offset]));
  412. VERIFY(test::areApprox(ref, data2, PacketSize) && "internal::pset1");
  413. }
  414. {
  415. for (int i = 0; i < PacketSize * 4; ++i) ref[i] = data1[i / PacketSize];
  416. Packet A0, A1, A2, A3;
  417. internal::pbroadcast4<Packet>(data1, A0, A1, A2, A3);
  418. internal::pstore(data2 + 0 * PacketSize, A0);
  419. internal::pstore(data2 + 1 * PacketSize, A1);
  420. internal::pstore(data2 + 2 * PacketSize, A2);
  421. internal::pstore(data2 + 3 * PacketSize, A3);
  422. VERIFY(test::areApprox(ref, data2, 4 * PacketSize) && "internal::pbroadcast4");
  423. }
  424. {
  425. for (int i = 0; i < PacketSize * 2; ++i) ref[i] = data1[i / PacketSize];
  426. Packet A0, A1;
  427. internal::pbroadcast2<Packet>(data1, A0, A1);
  428. internal::pstore(data2 + 0 * PacketSize, A0);
  429. internal::pstore(data2 + 1 * PacketSize, A1);
  430. VERIFY(test::areApprox(ref, data2, 2 * PacketSize) && "internal::pbroadcast2");
  431. }
  432. VERIFY(internal::isApprox(data1[0], internal::pfirst(internal::pload<Packet>(data1))) && "internal::pfirst");
  433. if (PacketSize > 1) {
  434. // apply different offsets to check that ploaddup is robust to unaligned inputs
  435. for (int offset = 0; offset < 4; ++offset) {
  436. for (int i = 0; i < PacketSize / 2; ++i) ref[2 * i + 0] = ref[2 * i + 1] = data1[offset + i];
  437. internal::pstore(data2, internal::ploaddup<Packet>(data1 + offset));
  438. VERIFY(test::areApprox(ref, data2, PacketSize) && "ploaddup");
  439. }
  440. }
  441. if (PacketSize > 2) {
  442. // apply different offsets to check that ploadquad is robust to unaligned inputs
  443. for (int offset = 0; offset < 4; ++offset) {
  444. for (int i = 0; i < PacketSize / 4; ++i)
  445. ref[4 * i + 0] = ref[4 * i + 1] = ref[4 * i + 2] = ref[4 * i + 3] = data1[offset + i];
  446. internal::pstore(data2, internal::ploadquad<Packet>(data1 + offset));
  447. VERIFY(test::areApprox(ref, data2, PacketSize) && "ploadquad");
  448. }
  449. }
  450. ref[0] = Scalar(0);
  451. for (int i = 0; i < PacketSize; ++i) ref[0] += data1[i];
  452. VERIFY(test::isApproxAbs(ref[0], internal::predux(internal::pload<Packet>(data1)), refvalue) && "internal::predux");
  453. if (!internal::is_same<Packet, typename internal::unpacket_traits<Packet>::half>::value) {
  454. int HalfPacketSize = PacketSize > 4 ? PacketSize / 2 : PacketSize;
  455. for (int i = 0; i < HalfPacketSize; ++i) ref[i] = Scalar(0);
  456. for (int i = 0; i < PacketSize; ++i) ref[i % HalfPacketSize] += data1[i];
  457. internal::pstore(data2, internal::predux_half_dowto4(internal::pload<Packet>(data1)));
  458. VERIFY(test::areApprox(ref, data2, HalfPacketSize) && "internal::predux_half_dowto4");
  459. }
  460. ref[0] = Scalar(1);
  461. for (int i = 0; i < PacketSize; ++i) ref[0] = REF_MUL(ref[0], data1[i]);
  462. VERIFY(internal::isApprox(ref[0], internal::predux_mul(internal::pload<Packet>(data1))) && "internal::predux_mul");
  463. for (int i = 0; i < PacketSize; ++i) ref[i] = data1[PacketSize - i - 1];
  464. internal::pstore(data2, internal::preverse(internal::pload<Packet>(data1)));
  465. VERIFY(test::areApprox(ref, data2, PacketSize) && "internal::preverse");
  466. internal::PacketBlock<Packet> kernel;
  467. for (int i = 0; i < PacketSize; ++i) {
  468. kernel.packet[i] = internal::pload<Packet>(data1 + i * PacketSize);
  469. }
  470. ptranspose(kernel);
  471. for (int i = 0; i < PacketSize; ++i) {
  472. internal::pstore(data2, kernel.packet[i]);
  473. for (int j = 0; j < PacketSize; ++j) {
  474. VERIFY(test::isApproxAbs(data2[j], data1[i + j * PacketSize], refvalue) && "ptranspose");
  475. }
  476. }
  477. // GeneralBlockPanelKernel also checks PacketBlock<Packet,(PacketSize%4)==0?4:PacketSize>;
  478. if (PacketSize > 4 && PacketSize % 4 == 0) {
  479. internal::PacketBlock<Packet, PacketSize%4==0?4:PacketSize> kernel2;
  480. for (int i = 0; i < 4; ++i) {
  481. kernel2.packet[i] = internal::pload<Packet>(data1 + i * PacketSize);
  482. }
  483. ptranspose(kernel2);
  484. int data_counter = 0;
  485. for (int i = 0; i < PacketSize; ++i) {
  486. for (int j = 0; j < 4; ++j) {
  487. data2[data_counter++] = data1[j*PacketSize + i];
  488. }
  489. }
  490. for (int i = 0; i < 4; ++i) {
  491. internal::pstore(data3, kernel2.packet[i]);
  492. for (int j = 0; j < PacketSize; ++j) {
  493. VERIFY(test::isApproxAbs(data3[j], data2[i*PacketSize + j], refvalue) && "ptranspose");
  494. }
  495. }
  496. }
  497. if (PacketTraits::HasBlend) {
  498. Packet thenPacket = internal::pload<Packet>(data1);
  499. Packet elsePacket = internal::pload<Packet>(data2);
  500. EIGEN_ALIGN_MAX internal::Selector<PacketSize> selector;
  501. for (int i = 0; i < PacketSize; ++i) {
  502. selector.select[i] = i;
  503. }
  504. Packet blend = internal::pblend(selector, thenPacket, elsePacket);
  505. EIGEN_ALIGN_MAX Scalar result[size];
  506. internal::pstore(result, blend);
  507. for (int i = 0; i < PacketSize; ++i) {
  508. VERIFY(test::isApproxAbs(result[i], (selector.select[i] ? data1[i] : data2[i]), refvalue));
  509. }
  510. }
  511. {
  512. for (int i = 0; i < PacketSize; ++i) {
  513. // "if" mask
  514. unsigned char v = internal::random<bool>() ? 0xff : 0;
  515. char* bytes = (char*)(data1 + i);
  516. for (int k = 0; k < int(sizeof(Scalar)); ++k) {
  517. bytes[k] = v;
  518. }
  519. // "then" packet
  520. data1[i + PacketSize] = internal::random<Scalar>();
  521. // "else" packet
  522. data1[i + 2 * PacketSize] = internal::random<Scalar>();
  523. }
  524. CHECK_CWISE3_IF(true, internal::pselect, internal::pselect);
  525. }
  526. for (int i = 0; i < size; ++i) {
  527. data1[i] = internal::random<Scalar>();
  528. }
  529. CHECK_CWISE1(internal::pzero, internal::pzero);
  530. CHECK_CWISE2_IF(true, internal::por, internal::por);
  531. CHECK_CWISE2_IF(true, internal::pxor, internal::pxor);
  532. CHECK_CWISE2_IF(true, internal::pand, internal::pand);
  533. packetmath_boolean_mask_ops<Scalar, Packet>();
  534. packetmath_pcast_ops_runner<Scalar, Packet>::run();
  535. packetmath_minus_zero_add<Scalar, Packet>();
  536. for (int i = 0; i < size; ++i) {
  537. data1[i] = numext::abs(internal::random<Scalar>());
  538. }
  539. CHECK_CWISE1_IF(PacketTraits::HasSqrt, numext::sqrt, internal::psqrt);
  540. CHECK_CWISE1_IF(PacketTraits::HasRsqrt, numext::rsqrt, internal::prsqrt);
  541. }
  542. // Notice that this definition works for complex types as well.
  543. // c++11 has std::log2 for real, but not for complex types.
  544. template <typename Scalar>
  545. Scalar log2(Scalar x) {
  546. return Scalar(EIGEN_LOG2E) * std::log(x);
  547. }
  548. template <typename Scalar, typename Packet>
  549. void packetmath_real() {
  550. typedef internal::packet_traits<Scalar> PacketTraits;
  551. const int PacketSize = internal::unpacket_traits<Packet>::size;
  552. const int size = PacketSize * 4;
  553. EIGEN_ALIGN_MAX Scalar data1[PacketSize * 4];
  554. EIGEN_ALIGN_MAX Scalar data2[PacketSize * 4];
  555. EIGEN_ALIGN_MAX Scalar ref[PacketSize * 4];
  556. for (int i = 0; i < size; ++i) {
  557. data1[i] = Scalar(internal::random<double>(0, 1) * std::pow(10., internal::random<double>(-6, 6)));
  558. data2[i] = Scalar(internal::random<double>(0, 1) * std::pow(10., internal::random<double>(-6, 6)));
  559. }
  560. if (internal::random<float>(0, 1) < 0.1f) data1[internal::random<int>(0, PacketSize)] = Scalar(0);
  561. CHECK_CWISE1_IF(PacketTraits::HasLog, std::log, internal::plog);
  562. CHECK_CWISE1_IF(PacketTraits::HasLog, log2, internal::plog2);
  563. CHECK_CWISE1_IF(PacketTraits::HasRsqrt, numext::rsqrt, internal::prsqrt);
  564. for (int i = 0; i < size; ++i) {
  565. data1[i] = Scalar(internal::random<double>(-1, 1) * std::pow(10., internal::random<double>(-3, 3)));
  566. data2[i] = Scalar(internal::random<double>(-1, 1) * std::pow(10., internal::random<double>(-3, 3)));
  567. }
  568. CHECK_CWISE1_IF(PacketTraits::HasSin, std::sin, internal::psin);
  569. CHECK_CWISE1_IF(PacketTraits::HasCos, std::cos, internal::pcos);
  570. CHECK_CWISE1_IF(PacketTraits::HasTan, std::tan, internal::ptan);
  571. CHECK_CWISE1_EXACT_IF(PacketTraits::HasRound, numext::round, internal::pround);
  572. CHECK_CWISE1_EXACT_IF(PacketTraits::HasCeil, numext::ceil, internal::pceil);
  573. CHECK_CWISE1_EXACT_IF(PacketTraits::HasFloor, numext::floor, internal::pfloor);
  574. CHECK_CWISE1_EXACT_IF(PacketTraits::HasRint, numext::rint, internal::print);
  575. packetmath_boolean_mask_ops_real<Scalar,Packet>();
  576. // Rounding edge cases.
  577. if (PacketTraits::HasRound || PacketTraits::HasCeil || PacketTraits::HasFloor || PacketTraits::HasRint) {
  578. typedef typename internal::make_integer<Scalar>::type IntType;
  579. // Start with values that cannot fit inside an integer, work down to less than one.
  580. Scalar val = numext::mini(
  581. Scalar(2) * static_cast<Scalar>(NumTraits<IntType>::highest()),
  582. NumTraits<Scalar>::highest());
  583. std::vector<Scalar> values;
  584. while (val > Scalar(0.25)) {
  585. // Cover both even and odd, positive and negative cases.
  586. values.push_back(val);
  587. values.push_back(val + Scalar(0.3));
  588. values.push_back(val + Scalar(0.5));
  589. values.push_back(val + Scalar(0.8));
  590. values.push_back(val + Scalar(1));
  591. values.push_back(val + Scalar(1.3));
  592. values.push_back(val + Scalar(1.5));
  593. values.push_back(val + Scalar(1.8));
  594. values.push_back(-val);
  595. values.push_back(-val - Scalar(0.3));
  596. values.push_back(-val - Scalar(0.5));
  597. values.push_back(-val - Scalar(0.8));
  598. values.push_back(-val - Scalar(1));
  599. values.push_back(-val - Scalar(1.3));
  600. values.push_back(-val - Scalar(1.5));
  601. values.push_back(-val - Scalar(1.8));
  602. values.push_back(Scalar(-1.5) + val); // Bug 1785.
  603. val = val / Scalar(2);
  604. }
  605. values.push_back(NumTraits<Scalar>::infinity());
  606. values.push_back(-NumTraits<Scalar>::infinity());
  607. values.push_back(NumTraits<Scalar>::quiet_NaN());
  608. for (size_t k=0; k<values.size(); ++k) {
  609. data1[0] = values[k];
  610. CHECK_CWISE1_EXACT_IF(PacketTraits::HasRound, numext::round, internal::pround);
  611. CHECK_CWISE1_EXACT_IF(PacketTraits::HasCeil, numext::ceil, internal::pceil);
  612. CHECK_CWISE1_EXACT_IF(PacketTraits::HasFloor, numext::floor, internal::pfloor);
  613. CHECK_CWISE1_EXACT_IF(PacketTraits::HasRint, numext::rint, internal::print);
  614. }
  615. }
  616. for (int i = 0; i < size; ++i) {
  617. data1[i] = Scalar(internal::random<double>(-1, 1));
  618. data2[i] = Scalar(internal::random<double>(-1, 1));
  619. }
  620. CHECK_CWISE1_IF(PacketTraits::HasASin, std::asin, internal::pasin);
  621. CHECK_CWISE1_IF(PacketTraits::HasACos, std::acos, internal::pacos);
  622. for (int i = 0; i < size; ++i) {
  623. data1[i] = Scalar(internal::random<double>(-87, 88));
  624. data2[i] = Scalar(internal::random<double>(-87, 88));
  625. }
  626. CHECK_CWISE1_IF(PacketTraits::HasExp, std::exp, internal::pexp);
  627. CHECK_CWISE1_BYREF1_IF(PacketTraits::HasExp, REF_FREXP, internal::pfrexp);
  628. if (PacketTraits::HasExp) {
  629. // Check denormals:
  630. for (int j=0; j<3; ++j) {
  631. data1[0] = Scalar(std::ldexp(1, NumTraits<Scalar>::min_exponent()-j));
  632. CHECK_CWISE1_BYREF1_IF(PacketTraits::HasExp, REF_FREXP, internal::pfrexp);
  633. data1[0] = -data1[0];
  634. CHECK_CWISE1_BYREF1_IF(PacketTraits::HasExp, REF_FREXP, internal::pfrexp);
  635. }
  636. // zero
  637. data1[0] = Scalar(0);
  638. CHECK_CWISE1_BYREF1_IF(PacketTraits::HasExp, REF_FREXP, internal::pfrexp);
  639. // inf and NaN only compare output fraction, not exponent.
  640. test::packet_helper<PacketTraits::HasExp,Packet> h;
  641. Packet pout;
  642. Scalar sout;
  643. Scalar special[] = { NumTraits<Scalar>::infinity(),
  644. -NumTraits<Scalar>::infinity(),
  645. NumTraits<Scalar>::quiet_NaN()};
  646. for (int i=0; i<3; ++i) {
  647. data1[0] = special[i];
  648. ref[0] = Scalar(REF_FREXP(data1[0], ref[PacketSize]));
  649. h.store(data2, internal::pfrexp(h.load(data1), h.forward_reference(pout, sout)));
  650. VERIFY(test::areApprox(ref, data2, 1) && "internal::pfrexp");
  651. }
  652. }
  653. for (int i = 0; i < PacketSize; ++i) {
  654. data1[i] = Scalar(internal::random<double>(-1, 1));
  655. data2[i] = Scalar(internal::random<double>(-1, 1));
  656. }
  657. for (int i = 0; i < PacketSize; ++i) {
  658. data1[i+PacketSize] = Scalar(internal::random<int>(-4, 4));
  659. data2[i+PacketSize] = Scalar(internal::random<double>(-4, 4));
  660. }
  661. CHECK_CWISE2_IF(PacketTraits::HasExp, REF_LDEXP, internal::pldexp);
  662. if (PacketTraits::HasExp) {
  663. data1[0] = Scalar(-1);
  664. // underflow to zero
  665. data1[PacketSize] = Scalar(NumTraits<Scalar>::min_exponent()-55);
  666. CHECK_CWISE2_IF(PacketTraits::HasExp, REF_LDEXP, internal::pldexp);
  667. // overflow to inf
  668. data1[PacketSize] = Scalar(NumTraits<Scalar>::max_exponent()+10);
  669. CHECK_CWISE2_IF(PacketTraits::HasExp, REF_LDEXP, internal::pldexp);
  670. // NaN stays NaN
  671. data1[0] = NumTraits<Scalar>::quiet_NaN();
  672. CHECK_CWISE2_IF(PacketTraits::HasExp, REF_LDEXP, internal::pldexp);
  673. VERIFY((numext::isnan)(data2[0]));
  674. // inf stays inf
  675. data1[0] = NumTraits<Scalar>::infinity();
  676. data1[PacketSize] = Scalar(NumTraits<Scalar>::min_exponent()-10);
  677. CHECK_CWISE2_IF(PacketTraits::HasExp, REF_LDEXP, internal::pldexp);
  678. // zero stays zero
  679. data1[0] = Scalar(0);
  680. data1[PacketSize] = Scalar(NumTraits<Scalar>::max_exponent()+10);
  681. CHECK_CWISE2_IF(PacketTraits::HasExp, REF_LDEXP, internal::pldexp);
  682. // Small number big exponent.
  683. data1[0] = Scalar(std::ldexp(Scalar(1.0), NumTraits<Scalar>::min_exponent()-1));
  684. data1[PacketSize] = Scalar(-NumTraits<Scalar>::min_exponent()
  685. +NumTraits<Scalar>::max_exponent());
  686. CHECK_CWISE2_IF(PacketTraits::HasExp, REF_LDEXP, internal::pldexp);
  687. // Big number small exponent.
  688. data1[0] = Scalar(std::ldexp(Scalar(1.0), NumTraits<Scalar>::max_exponent()-1));
  689. data1[PacketSize] = Scalar(+NumTraits<Scalar>::min_exponent()
  690. -NumTraits<Scalar>::max_exponent());
  691. CHECK_CWISE2_IF(PacketTraits::HasExp, REF_LDEXP, internal::pldexp);
  692. }
  693. for (int i = 0; i < size; ++i) {
  694. data1[i] = Scalar(internal::random<double>(-1, 1) * std::pow(10., internal::random<double>(-6, 6)));
  695. data2[i] = Scalar(internal::random<double>(-1, 1) * std::pow(10., internal::random<double>(-6, 6)));
  696. }
  697. data1[0] = Scalar(1e-20);
  698. CHECK_CWISE1_IF(PacketTraits::HasTanh, std::tanh, internal::ptanh);
  699. if (PacketTraits::HasExp && PacketSize >= 2) {
  700. const Scalar small = NumTraits<Scalar>::epsilon();
  701. data1[0] = NumTraits<Scalar>::quiet_NaN();
  702. data1[1] = small;
  703. test::packet_helper<PacketTraits::HasExp, Packet> h;
  704. h.store(data2, internal::pexp(h.load(data1)));
  705. VERIFY((numext::isnan)(data2[0]));
  706. // TODO(rmlarsen): Re-enable for bfloat16.
  707. if (!internal::is_same<Scalar, bfloat16>::value) {
  708. VERIFY_IS_APPROX(std::exp(small), data2[1]);
  709. }
  710. data1[0] = -small;
  711. data1[1] = Scalar(0);
  712. h.store(data2, internal::pexp(h.load(data1)));
  713. // TODO(rmlarsen): Re-enable for bfloat16.
  714. if (!internal::is_same<Scalar, bfloat16>::value) {
  715. VERIFY_IS_APPROX(std::exp(-small), data2[0]);
  716. }
  717. VERIFY_IS_EQUAL(std::exp(Scalar(0)), data2[1]);
  718. data1[0] = (std::numeric_limits<Scalar>::min)();
  719. data1[1] = -(std::numeric_limits<Scalar>::min)();
  720. h.store(data2, internal::pexp(h.load(data1)));
  721. VERIFY_IS_APPROX(std::exp((std::numeric_limits<Scalar>::min)()), data2[0]);
  722. VERIFY_IS_APPROX(std::exp(-(std::numeric_limits<Scalar>::min)()), data2[1]);
  723. data1[0] = std::numeric_limits<Scalar>::denorm_min();
  724. data1[1] = -std::numeric_limits<Scalar>::denorm_min();
  725. h.store(data2, internal::pexp(h.load(data1)));
  726. VERIFY_IS_APPROX(std::exp(std::numeric_limits<Scalar>::denorm_min()), data2[0]);
  727. VERIFY_IS_APPROX(std::exp(-std::numeric_limits<Scalar>::denorm_min()), data2[1]);
  728. }
  729. if (PacketTraits::HasTanh) {
  730. // NOTE this test migh fail with GCC prior to 6.3, see MathFunctionsImpl.h for details.
  731. data1[0] = NumTraits<Scalar>::quiet_NaN();
  732. test::packet_helper<internal::packet_traits<Scalar>::HasTanh, Packet> h;
  733. h.store(data2, internal::ptanh(h.load(data1)));
  734. VERIFY((numext::isnan)(data2[0]));
  735. }
  736. if (PacketTraits::HasExp) {
  737. internal::scalar_logistic_op<Scalar> logistic;
  738. for (int i = 0; i < size; ++i) {
  739. data1[i] = Scalar(internal::random<double>(-20, 20));
  740. }
  741. test::packet_helper<PacketTraits::HasExp, Packet> h;
  742. h.store(data2, logistic.packetOp(h.load(data1)));
  743. for (int i = 0; i < PacketSize; ++i) {
  744. VERIFY_IS_APPROX(data2[i], logistic(data1[i]));
  745. }
  746. }
  747. #if EIGEN_HAS_C99_MATH && (EIGEN_COMP_CXXVER >= 11)
  748. data1[0] = NumTraits<Scalar>::infinity();
  749. data1[1] = Scalar(-1);
  750. CHECK_CWISE1_IF(PacketTraits::HasLog1p, std::log1p, internal::plog1p);
  751. data1[0] = NumTraits<Scalar>::infinity();
  752. data1[1] = -NumTraits<Scalar>::infinity();
  753. CHECK_CWISE1_IF(PacketTraits::HasExpm1, std::expm1, internal::pexpm1);
  754. #endif
  755. if (PacketSize >= 2) {
  756. data1[0] = NumTraits<Scalar>::quiet_NaN();
  757. data1[1] = NumTraits<Scalar>::epsilon();
  758. if (PacketTraits::HasLog) {
  759. test::packet_helper<PacketTraits::HasLog, Packet> h;
  760. h.store(data2, internal::plog(h.load(data1)));
  761. VERIFY((numext::isnan)(data2[0]));
  762. // TODO(cantonios): Re-enable for bfloat16.
  763. if (!internal::is_same<Scalar, bfloat16>::value) {
  764. VERIFY_IS_APPROX(std::log(data1[1]), data2[1]);
  765. }
  766. data1[0] = -NumTraits<Scalar>::epsilon();
  767. data1[1] = Scalar(0);
  768. h.store(data2, internal::plog(h.load(data1)));
  769. VERIFY((numext::isnan)(data2[0]));
  770. VERIFY_IS_EQUAL(std::log(Scalar(0)), data2[1]);
  771. data1[0] = (std::numeric_limits<Scalar>::min)();
  772. data1[1] = -(std::numeric_limits<Scalar>::min)();
  773. h.store(data2, internal::plog(h.load(data1)));
  774. // TODO(cantonios): Re-enable for bfloat16.
  775. if (!internal::is_same<Scalar, bfloat16>::value) {
  776. VERIFY_IS_APPROX(std::log((std::numeric_limits<Scalar>::min)()), data2[0]);
  777. }
  778. VERIFY((numext::isnan)(data2[1]));
  779. // Note: 32-bit arm always flushes denorms to zero.
  780. #if !EIGEN_ARCH_ARM
  781. if (std::numeric_limits<Scalar>::has_denorm == std::denorm_present) {
  782. data1[0] = std::numeric_limits<Scalar>::denorm_min();
  783. data1[1] = -std::numeric_limits<Scalar>::denorm_min();
  784. h.store(data2, internal::plog(h.load(data1)));
  785. // TODO(rmlarsen): Reenable.
  786. // VERIFY_IS_EQUAL(std::log(std::numeric_limits<Scalar>::denorm_min()), data2[0]);
  787. VERIFY((numext::isnan)(data2[1]));
  788. }
  789. #endif
  790. data1[0] = Scalar(-1.0f);
  791. h.store(data2, internal::plog(h.load(data1)));
  792. VERIFY((numext::isnan)(data2[0]));
  793. data1[0] = NumTraits<Scalar>::infinity();
  794. h.store(data2, internal::plog(h.load(data1)));
  795. VERIFY((numext::isinf)(data2[0]));
  796. }
  797. if (PacketTraits::HasLog1p) {
  798. test::packet_helper<PacketTraits::HasLog1p, Packet> h;
  799. data1[0] = Scalar(-2);
  800. data1[1] = -NumTraits<Scalar>::infinity();
  801. h.store(data2, internal::plog1p(h.load(data1)));
  802. VERIFY((numext::isnan)(data2[0]));
  803. VERIFY((numext::isnan)(data2[1]));
  804. }
  805. if (PacketTraits::HasSqrt) {
  806. test::packet_helper<PacketTraits::HasSqrt, Packet> h;
  807. data1[0] = Scalar(-1.0f);
  808. if (std::numeric_limits<Scalar>::has_denorm == std::denorm_present) {
  809. data1[1] = -std::numeric_limits<Scalar>::denorm_min();
  810. } else {
  811. data1[1] = -NumTraits<Scalar>::epsilon();
  812. }
  813. h.store(data2, internal::psqrt(h.load(data1)));
  814. VERIFY((numext::isnan)(data2[0]));
  815. VERIFY((numext::isnan)(data2[1]));
  816. }
  817. // TODO(rmlarsen): Re-enable for half and bfloat16.
  818. if (PacketTraits::HasCos
  819. && !internal::is_same<Scalar, half>::value
  820. && !internal::is_same<Scalar, bfloat16>::value) {
  821. test::packet_helper<PacketTraits::HasCos, Packet> h;
  822. for (Scalar k = Scalar(1); k < Scalar(10000) / NumTraits<Scalar>::epsilon(); k *= Scalar(2)) {
  823. for (int k1 = 0; k1 <= 1; ++k1) {
  824. data1[0] = Scalar((2 * double(k) + k1) * double(EIGEN_PI) / 2 * internal::random<double>(0.8, 1.2));
  825. data1[1] = Scalar((2 * double(k) + 2 + k1) * double(EIGEN_PI) / 2 * internal::random<double>(0.8, 1.2));
  826. h.store(data2, internal::pcos(h.load(data1)));
  827. h.store(data2 + PacketSize, internal::psin(h.load(data1)));
  828. VERIFY(data2[0] <= Scalar(1.) && data2[0] >= Scalar(-1.));
  829. VERIFY(data2[1] <= Scalar(1.) && data2[1] >= Scalar(-1.));
  830. VERIFY(data2[PacketSize + 0] <= Scalar(1.) && data2[PacketSize + 0] >= Scalar(-1.));
  831. VERIFY(data2[PacketSize + 1] <= Scalar(1.) && data2[PacketSize + 1] >= Scalar(-1.));
  832. VERIFY_IS_APPROX(data2[0], std::cos(data1[0]));
  833. VERIFY_IS_APPROX(data2[1], std::cos(data1[1]));
  834. VERIFY_IS_APPROX(data2[PacketSize + 0], std::sin(data1[0]));
  835. VERIFY_IS_APPROX(data2[PacketSize + 1], std::sin(data1[1]));
  836. VERIFY_IS_APPROX(numext::abs2(data2[0]) + numext::abs2(data2[PacketSize + 0]), Scalar(1));
  837. VERIFY_IS_APPROX(numext::abs2(data2[1]) + numext::abs2(data2[PacketSize + 1]), Scalar(1));
  838. }
  839. }
  840. data1[0] = NumTraits<Scalar>::infinity();
  841. data1[1] = -NumTraits<Scalar>::infinity();
  842. h.store(data2, internal::psin(h.load(data1)));
  843. VERIFY((numext::isnan)(data2[0]));
  844. VERIFY((numext::isnan)(data2[1]));
  845. h.store(data2, internal::pcos(h.load(data1)));
  846. VERIFY((numext::isnan)(data2[0]));
  847. VERIFY((numext::isnan)(data2[1]));
  848. data1[0] = NumTraits<Scalar>::quiet_NaN();
  849. h.store(data2, internal::psin(h.load(data1)));
  850. VERIFY((numext::isnan)(data2[0]));
  851. h.store(data2, internal::pcos(h.load(data1)));
  852. VERIFY((numext::isnan)(data2[0]));
  853. data1[0] = -Scalar(0.);
  854. h.store(data2, internal::psin(h.load(data1)));
  855. VERIFY(internal::biteq(data2[0], data1[0]));
  856. h.store(data2, internal::pcos(h.load(data1)));
  857. VERIFY_IS_EQUAL(data2[0], Scalar(1));
  858. }
  859. }
  860. }
  861. #define CAST_CHECK_CWISE1_IF(COND, REFOP, POP, SCALAR, REFTYPE) if(COND) { \
  862. test::packet_helper<COND,Packet> h; \
  863. for (int i=0; i<PacketSize; ++i) \
  864. ref[i] = SCALAR(REFOP(static_cast<REFTYPE>(data1[i]))); \
  865. h.store(data2, POP(h.load(data1))); \
  866. VERIFY(test::areApprox(ref, data2, PacketSize) && #POP); \
  867. }
  868. template <typename Scalar>
  869. Scalar propagate_nan_max(const Scalar& a, const Scalar& b) {
  870. if ((numext::isnan)(a)) return a;
  871. if ((numext::isnan)(b)) return b;
  872. return (numext::maxi)(a,b);
  873. }
  874. template <typename Scalar>
  875. Scalar propagate_nan_min(const Scalar& a, const Scalar& b) {
  876. if ((numext::isnan)(a)) return a;
  877. if ((numext::isnan)(b)) return b;
  878. return (numext::mini)(a,b);
  879. }
  880. template <typename Scalar>
  881. Scalar propagate_number_max(const Scalar& a, const Scalar& b) {
  882. if ((numext::isnan)(a)) return b;
  883. if ((numext::isnan)(b)) return a;
  884. return (numext::maxi)(a,b);
  885. }
  886. template <typename Scalar>
  887. Scalar propagate_number_min(const Scalar& a, const Scalar& b) {
  888. if ((numext::isnan)(a)) return b;
  889. if ((numext::isnan)(b)) return a;
  890. return (numext::mini)(a,b);
  891. }
  892. template <typename Scalar, typename Packet>
  893. void packetmath_notcomplex() {
  894. typedef internal::packet_traits<Scalar> PacketTraits;
  895. const int PacketSize = internal::unpacket_traits<Packet>::size;
  896. EIGEN_ALIGN_MAX Scalar data1[PacketSize * 4];
  897. EIGEN_ALIGN_MAX Scalar data2[PacketSize * 4];
  898. EIGEN_ALIGN_MAX Scalar ref[PacketSize * 4];
  899. Array<Scalar, Dynamic, 1>::Map(data1, PacketSize * 4).setRandom();
  900. VERIFY((!PacketTraits::Vectorizable) || PacketTraits::HasMin);
  901. VERIFY((!PacketTraits::Vectorizable) || PacketTraits::HasMax);
  902. CHECK_CWISE2_IF(PacketTraits::HasMin, (std::min), internal::pmin);
  903. CHECK_CWISE2_IF(PacketTraits::HasMax, (std::max), internal::pmax);
  904. CHECK_CWISE2_IF(PacketTraits::HasMin, propagate_number_min, internal::pmin<PropagateNumbers>);
  905. CHECK_CWISE2_IF(PacketTraits::HasMax, propagate_number_max, internal::pmax<PropagateNumbers>);
  906. CHECK_CWISE1(numext::abs, internal::pabs);
  907. CHECK_CWISE2_IF(PacketTraits::HasAbsDiff, REF_ABS_DIFF, internal::pabsdiff);
  908. ref[0] = data1[0];
  909. for (int i = 0; i < PacketSize; ++i) ref[0] = internal::pmin(ref[0], data1[i]);
  910. VERIFY(internal::isApprox(ref[0], internal::predux_min(internal::pload<Packet>(data1))) && "internal::predux_min");
  911. ref[0] = data1[0];
  912. for (int i = 0; i < PacketSize; ++i) ref[0] = internal::pmax(ref[0], data1[i]);
  913. VERIFY(internal::isApprox(ref[0], internal::predux_max(internal::pload<Packet>(data1))) && "internal::predux_max");
  914. for (int i = 0; i < PacketSize; ++i) ref[i] = data1[0] + Scalar(i);
  915. internal::pstore(data2, internal::plset<Packet>(data1[0]));
  916. VERIFY(test::areApprox(ref, data2, PacketSize) && "internal::plset");
  917. {
  918. unsigned char* data1_bits = reinterpret_cast<unsigned char*>(data1);
  919. // predux_all - not needed yet
  920. // for (unsigned int i=0; i<PacketSize*sizeof(Scalar); ++i) data1_bits[i] = 0xff;
  921. // VERIFY(internal::predux_all(internal::pload<Packet>(data1)) && "internal::predux_all(1111)");
  922. // for(int k=0; k<PacketSize; ++k)
  923. // {
  924. // for (unsigned int i=0; i<sizeof(Scalar); ++i) data1_bits[k*sizeof(Scalar)+i] = 0x0;
  925. // VERIFY( (!internal::predux_all(internal::pload<Packet>(data1))) && "internal::predux_all(0101)");
  926. // for (unsigned int i=0; i<sizeof(Scalar); ++i) data1_bits[k*sizeof(Scalar)+i] = 0xff;
  927. // }
  928. // predux_any
  929. for (unsigned int i = 0; i < PacketSize * sizeof(Scalar); ++i) data1_bits[i] = 0x0;
  930. VERIFY((!internal::predux_any(internal::pload<Packet>(data1))) && "internal::predux_any(0000)");
  931. for (int k = 0; k < PacketSize; ++k) {
  932. for (unsigned int i = 0; i < sizeof(Scalar); ++i) data1_bits[k * sizeof(Scalar) + i] = 0xff;
  933. VERIFY(internal::predux_any(internal::pload<Packet>(data1)) && "internal::predux_any(0101)");
  934. for (unsigned int i = 0; i < sizeof(Scalar); ++i) data1_bits[k * sizeof(Scalar) + i] = 0x00;
  935. }
  936. }
  937. // Test NaN propagation.
  938. if (!NumTraits<Scalar>::IsInteger) {
  939. // Test reductions with no NaNs.
  940. ref[0] = data1[0];
  941. for (int i = 0; i < PacketSize; ++i) ref[0] = internal::pmin<PropagateNumbers>(ref[0], data1[i]);
  942. VERIFY(internal::isApprox(ref[0], internal::predux_min<PropagateNumbers>(internal::pload<Packet>(data1))) && "internal::predux_min<PropagateNumbers>");
  943. ref[0] = data1[0];
  944. for (int i = 0; i < PacketSize; ++i) ref[0] = internal::pmin<PropagateNaN>(ref[0], data1[i]);
  945. VERIFY(internal::isApprox(ref[0], internal::predux_min<PropagateNaN>(internal::pload<Packet>(data1))) && "internal::predux_min<PropagateNaN>");
  946. ref[0] = data1[0];
  947. for (int i = 0; i < PacketSize; ++i) ref[0] = internal::pmax<PropagateNumbers>(ref[0], data1[i]);
  948. VERIFY(internal::isApprox(ref[0], internal::predux_max<PropagateNumbers>(internal::pload<Packet>(data1))) && "internal::predux_max<PropagateNumbers>");
  949. ref[0] = data1[0];
  950. for (int i = 0; i < PacketSize; ++i) ref[0] = internal::pmax<PropagateNaN>(ref[0], data1[i]);
  951. VERIFY(internal::isApprox(ref[0], internal::predux_max<PropagateNaN>(internal::pload<Packet>(data1))) && "internal::predux_max<PropagateNumbers>");
  952. // A single NaN.
  953. const size_t index = std::numeric_limits<size_t>::quiet_NaN() % PacketSize;
  954. data1[index] = NumTraits<Scalar>::quiet_NaN();
  955. VERIFY(PacketSize==1 || !(numext::isnan)(internal::predux_min<PropagateNumbers>(internal::pload<Packet>(data1))));
  956. VERIFY((numext::isnan)(internal::predux_min<PropagateNaN>(internal::pload<Packet>(data1))));
  957. VERIFY(PacketSize==1 || !(numext::isnan)(internal::predux_max<PropagateNumbers>(internal::pload<Packet>(data1))));
  958. VERIFY((numext::isnan)(internal::predux_max<PropagateNaN>(internal::pload<Packet>(data1))));
  959. // All NaNs.
  960. for (int i = 0; i < 4 * PacketSize; ++i) data1[i] = NumTraits<Scalar>::quiet_NaN();
  961. VERIFY((numext::isnan)(internal::predux_min<PropagateNumbers>(internal::pload<Packet>(data1))));
  962. VERIFY((numext::isnan)(internal::predux_min<PropagateNaN>(internal::pload<Packet>(data1))));
  963. VERIFY((numext::isnan)(internal::predux_max<PropagateNumbers>(internal::pload<Packet>(data1))));
  964. VERIFY((numext::isnan)(internal::predux_max<PropagateNaN>(internal::pload<Packet>(data1))));
  965. // Test NaN propagation for coefficient-wise min and max.
  966. for (int i = 0; i < PacketSize; ++i) {
  967. data1[i] = internal::random<bool>() ? NumTraits<Scalar>::quiet_NaN() : Scalar(0);
  968. data1[i + PacketSize] = internal::random<bool>() ? NumTraits<Scalar>::quiet_NaN() : Scalar(0);
  969. }
  970. // Note: NaN propagation is implementation defined for pmin/pmax, so we do not test it here.
  971. CHECK_CWISE2_IF(PacketTraits::HasMin, propagate_number_min, (internal::pmin<PropagateNumbers>));
  972. CHECK_CWISE2_IF(PacketTraits::HasMax, propagate_number_max, internal::pmax<PropagateNumbers>);
  973. CHECK_CWISE2_IF(PacketTraits::HasMin, propagate_nan_min, (internal::pmin<PropagateNaN>));
  974. CHECK_CWISE2_IF(PacketTraits::HasMax, propagate_nan_max, internal::pmax<PropagateNaN>);
  975. }
  976. packetmath_boolean_mask_ops_notcomplex<Scalar, Packet>();
  977. }
  978. template <typename Scalar, typename Packet, bool ConjLhs, bool ConjRhs>
  979. void test_conj_helper(Scalar* data1, Scalar* data2, Scalar* ref, Scalar* pval) {
  980. const int PacketSize = internal::unpacket_traits<Packet>::size;
  981. internal::conj_if<ConjLhs> cj0;
  982. internal::conj_if<ConjRhs> cj1;
  983. internal::conj_helper<Scalar, Scalar, ConjLhs, ConjRhs> cj;
  984. internal::conj_helper<Packet, Packet, ConjLhs, ConjRhs> pcj;
  985. for (int i = 0; i < PacketSize; ++i) {
  986. ref[i] = cj0(data1[i]) * cj1(data2[i]);
  987. VERIFY(internal::isApprox(ref[i], cj.pmul(data1[i], data2[i])) && "conj_helper pmul");
  988. }
  989. internal::pstore(pval, pcj.pmul(internal::pload<Packet>(data1), internal::pload<Packet>(data2)));
  990. VERIFY(test::areApprox(ref, pval, PacketSize) && "conj_helper pmul");
  991. for (int i = 0; i < PacketSize; ++i) {
  992. Scalar tmp = ref[i];
  993. ref[i] += cj0(data1[i]) * cj1(data2[i]);
  994. VERIFY(internal::isApprox(ref[i], cj.pmadd(data1[i], data2[i], tmp)) && "conj_helper pmadd");
  995. }
  996. internal::pstore(
  997. pval, pcj.pmadd(internal::pload<Packet>(data1), internal::pload<Packet>(data2), internal::pload<Packet>(pval)));
  998. VERIFY(test::areApprox(ref, pval, PacketSize) && "conj_helper pmadd");
  999. }
  1000. template <typename Scalar, typename Packet>
  1001. void packetmath_complex() {
  1002. typedef internal::packet_traits<Scalar> PacketTraits;
  1003. typedef typename Scalar::value_type RealScalar;
  1004. const int PacketSize = internal::unpacket_traits<Packet>::size;
  1005. const int size = PacketSize * 4;
  1006. EIGEN_ALIGN_MAX Scalar data1[PacketSize * 4];
  1007. EIGEN_ALIGN_MAX Scalar data2[PacketSize * 4];
  1008. EIGEN_ALIGN_MAX Scalar ref[PacketSize * 4];
  1009. EIGEN_ALIGN_MAX Scalar pval[PacketSize * 4];
  1010. for (int i = 0; i < size; ++i) {
  1011. data1[i] = internal::random<Scalar>() * Scalar(1e2);
  1012. data2[i] = internal::random<Scalar>() * Scalar(1e2);
  1013. }
  1014. test_conj_helper<Scalar, Packet, false, false>(data1, data2, ref, pval);
  1015. test_conj_helper<Scalar, Packet, false, true>(data1, data2, ref, pval);
  1016. test_conj_helper<Scalar, Packet, true, false>(data1, data2, ref, pval);
  1017. test_conj_helper<Scalar, Packet, true, true>(data1, data2, ref, pval);
  1018. // Test pcplxflip.
  1019. {
  1020. for (int i = 0; i < PacketSize; ++i) ref[i] = Scalar(std::imag(data1[i]), std::real(data1[i]));
  1021. internal::pstore(pval, internal::pcplxflip(internal::pload<Packet>(data1)));
  1022. VERIFY(test::areApprox(ref, pval, PacketSize) && "pcplxflip");
  1023. }
  1024. if (PacketTraits::HasSqrt) {
  1025. for (int i = 0; i < size; ++i) {
  1026. data1[i] = Scalar(internal::random<RealScalar>(), internal::random<RealScalar>());
  1027. }
  1028. CHECK_CWISE1_N(numext::sqrt, internal::psqrt, size);
  1029. // Test misc. corner cases.
  1030. const RealScalar zero = RealScalar(0);
  1031. const RealScalar one = RealScalar(1);
  1032. const RealScalar inf = std::numeric_limits<RealScalar>::infinity();
  1033. const RealScalar nan = std::numeric_limits<RealScalar>::quiet_NaN();
  1034. data1[0] = Scalar(zero, zero);
  1035. data1[1] = Scalar(-zero, zero);
  1036. data1[2] = Scalar(one, zero);
  1037. data1[3] = Scalar(zero, one);
  1038. CHECK_CWISE1_N(numext::sqrt, internal::psqrt, 4);
  1039. data1[0] = Scalar(-one, zero);
  1040. data1[1] = Scalar(zero, -one);
  1041. data1[2] = Scalar(one, one);
  1042. data1[3] = Scalar(-one, -one);
  1043. CHECK_CWISE1_N(numext::sqrt, internal::psqrt, 4);
  1044. data1[0] = Scalar(inf, zero);
  1045. data1[1] = Scalar(zero, inf);
  1046. data1[2] = Scalar(-inf, zero);
  1047. data1[3] = Scalar(zero, -inf);
  1048. CHECK_CWISE1_N(numext::sqrt, internal::psqrt, 4);
  1049. data1[0] = Scalar(inf, inf);
  1050. data1[1] = Scalar(-inf, inf);
  1051. data1[2] = Scalar(inf, -inf);
  1052. data1[3] = Scalar(-inf, -inf);
  1053. CHECK_CWISE1_N(numext::sqrt, internal::psqrt, 4);
  1054. data1[0] = Scalar(nan, zero);
  1055. data1[1] = Scalar(zero, nan);
  1056. data1[2] = Scalar(nan, one);
  1057. data1[3] = Scalar(one, nan);
  1058. CHECK_CWISE1_N(numext::sqrt, internal::psqrt, 4);
  1059. data1[0] = Scalar(nan, nan);
  1060. data1[1] = Scalar(inf, nan);
  1061. data1[2] = Scalar(nan, inf);
  1062. data1[3] = Scalar(-inf, nan);
  1063. CHECK_CWISE1_N(numext::sqrt, internal::psqrt, 4);
  1064. }
  1065. }
  1066. template <typename Scalar, typename Packet>
  1067. void packetmath_scatter_gather() {
  1068. typedef typename NumTraits<Scalar>::Real RealScalar;
  1069. const int PacketSize = internal::unpacket_traits<Packet>::size;
  1070. EIGEN_ALIGN_MAX Scalar data1[PacketSize];
  1071. RealScalar refvalue = RealScalar(0);
  1072. for (int i = 0; i < PacketSize; ++i) {
  1073. data1[i] = internal::random<Scalar>() / RealScalar(PacketSize);
  1074. }
  1075. int stride = internal::random<int>(1, 20);
  1076. // Buffer of zeros.
  1077. EIGEN_ALIGN_MAX Scalar buffer[PacketSize * 20] = {};
  1078. Packet packet = internal::pload<Packet>(data1);
  1079. internal::pscatter<Scalar, Packet>(buffer, packet, stride);
  1080. for (int i = 0; i < PacketSize * 20; ++i) {
  1081. if ((i % stride) == 0 && i < stride * PacketSize) {
  1082. VERIFY(test::isApproxAbs(buffer[i], data1[i / stride], refvalue) && "pscatter");
  1083. } else {
  1084. VERIFY(test::isApproxAbs(buffer[i], Scalar(0), refvalue) && "pscatter");
  1085. }
  1086. }
  1087. for (int i = 0; i < PacketSize * 7; ++i) {
  1088. buffer[i] = internal::random<Scalar>() / RealScalar(PacketSize);
  1089. }
  1090. packet = internal::pgather<Scalar, Packet>(buffer, 7);
  1091. internal::pstore(data1, packet);
  1092. for (int i = 0; i < PacketSize; ++i) {
  1093. VERIFY(test::isApproxAbs(data1[i], buffer[i * 7], refvalue) && "pgather");
  1094. }
  1095. }
  1096. namespace Eigen {
  1097. namespace test {
  1098. template <typename Scalar, typename PacketType>
  1099. struct runall<Scalar, PacketType, false, false> { // i.e. float or double
  1100. static void run() {
  1101. packetmath<Scalar, PacketType>();
  1102. packetmath_scatter_gather<Scalar, PacketType>();
  1103. packetmath_notcomplex<Scalar, PacketType>();
  1104. packetmath_real<Scalar, PacketType>();
  1105. }
  1106. };
  1107. template <typename Scalar, typename PacketType>
  1108. struct runall<Scalar, PacketType, false, true> { // i.e. int
  1109. static void run() {
  1110. packetmath<Scalar, PacketType>();
  1111. packetmath_scatter_gather<Scalar, PacketType>();
  1112. packetmath_notcomplex<Scalar, PacketType>();
  1113. }
  1114. };
  1115. template <typename Scalar, typename PacketType>
  1116. struct runall<Scalar, PacketType, true, false> { // i.e. complex
  1117. static void run() {
  1118. packetmath<Scalar, PacketType>();
  1119. packetmath_scatter_gather<Scalar, PacketType>();
  1120. packetmath_complex<Scalar, PacketType>();
  1121. }
  1122. };
  1123. } // namespace test
  1124. } // namespace Eigen
  1125. EIGEN_DECLARE_TEST(packetmath) {
  1126. g_first_pass = true;
  1127. for (int i = 0; i < g_repeat; i++) {
  1128. CALL_SUBTEST_1(test::runner<float>::run());
  1129. CALL_SUBTEST_2(test::runner<double>::run());
  1130. CALL_SUBTEST_3(test::runner<int8_t>::run());
  1131. CALL_SUBTEST_4(test::runner<uint8_t>::run());
  1132. CALL_SUBTEST_5(test::runner<int16_t>::run());
  1133. CALL_SUBTEST_6(test::runner<uint16_t>::run());
  1134. CALL_SUBTEST_7(test::runner<int32_t>::run());
  1135. CALL_SUBTEST_8(test::runner<uint32_t>::run());
  1136. CALL_SUBTEST_9(test::runner<int64_t>::run());
  1137. CALL_SUBTEST_10(test::runner<uint64_t>::run());
  1138. CALL_SUBTEST_11(test::runner<std::complex<float> >::run());
  1139. CALL_SUBTEST_12(test::runner<std::complex<double> >::run());
  1140. CALL_SUBTEST_13(test::runner<half>::run());
  1141. CALL_SUBTEST_14((packetmath<bool, internal::packet_traits<bool>::type>()));
  1142. CALL_SUBTEST_15(test::runner<bfloat16>::run());
  1143. g_first_pass = false;
  1144. }
  1145. }