blob: 62d3f60ff09e5d71ddf43b44150ff14f1296f6d3 [file] [log] [blame]
Gael Guennebaudcb71dc42009-01-07 22:20:03 +00001// This file is part of Eigen, a lightweight C++ template library
Benoit Jacob6347b1d2009-05-22 20:25:33 +02002// for linear algebra.
Gael Guennebaudcb71dc42009-01-07 22:20:03 +00003//
Gael Guennebaud836da912015-12-11 10:06:28 +01004// Copyright (C) 2015 Gael Guennebaud <gael.guennebaud@inria.fr>
Gael Guennebaudcb71dc42009-01-07 22:20:03 +00005//
Benoit Jacob69124cf2012-07-13 14:42:47 -04006// This Source Code Form is subject to the terms of the Mozilla
7// Public License v. 2.0. If a copy of the MPL was not distributed
8// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
Gael Guennebaudcb71dc42009-01-07 22:20:03 +00009
Gael Guennebaude68e1652016-05-24 21:54:03 +020010#ifdef EIGEN_TEST_PART_1
11#define EIGEN_UNALIGNED_VECTORIZE 1
12#endif
13
14#ifdef EIGEN_TEST_PART_2
15#define EIGEN_UNALIGNED_VECTORIZE 0
16#endif
17
Gael Guennebaud836da912015-12-11 10:06:28 +010018#ifdef EIGEN_DEFAULT_TO_ROW_MAJOR
19#undef EIGEN_DEFAULT_TO_ROW_MAJOR
20#endif
Benoit Jacob27f52502010-02-27 19:04:22 -050021#define EIGEN_DEBUG_ASSIGN
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000022#include "main.h"
23#include <typeinfo>
24
Gael Guennebaud2cf6d302018-09-20 11:38:19 +020025// Disable "ignoring attributes on template argument"
26// for packet_traits<Packet*>
27// => The only workaround would be to wrap _m128 and the likes
28// within wrappers.
29#if EIGEN_GNUC_AT_LEAST(6,0)
30 #pragma GCC diagnostic ignored "-Wignored-attributes"
31#endif
32
Gael Guennebaud0fc89542015-10-27 10:38:49 +010033using internal::demangle_flags;
34using internal::demangle_traversal;
35using internal::demangle_unrolling;
Gael Guennebaud51ec1882010-07-08 23:30:16 +020036
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000037template<typename Dst, typename Src>
Benoit Jacob94c706d2009-11-18 11:57:07 -050038bool test_assign(const Dst&, const Src&, int traversal, int unrolling)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000039{
Christoph Hertzberge3c82892018-09-21 21:15:51 +020040 EIGEN_STATIC_ASSERT_SAME_MATRIX_SIZE(Dst,Src);
Gael Guennebaud66e99ab2016-06-06 15:11:41 +020041 typedef internal::copy_using_evaluator_traits<internal::evaluator<Dst>,internal::evaluator<Src>, internal::assign_op<typename Dst::Scalar,typename Src::Scalar> > traits;
Gael Guennebaud512ba0a2016-04-13 18:16:35 +020042 bool res = traits::Traversal==traversal;
43 if(unrolling==InnerUnrolling+CompleteUnrolling)
Gael Guennebauda95e1a22016-04-13 22:00:38 +020044 res = res && (int(traits::Unrolling)==InnerUnrolling || int(traits::Unrolling)==CompleteUnrolling);
Gael Guennebaud512ba0a2016-04-13 18:16:35 +020045 else
Gael Guennebauda95e1a22016-04-13 22:00:38 +020046 res = res && int(traits::Unrolling)==unrolling;
Gael Guennebaud51ec1882010-07-08 23:30:16 +020047 if(!res)
48 {
Gael Guennebauda3950242014-03-12 18:14:58 +010049 std::cerr << "Src: " << demangle_flags(Src::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010050 std::cerr << " " << demangle_flags(internal::evaluator<Src>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010051 std::cerr << "Dst: " << demangle_flags(Dst::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010052 std::cerr << " " << demangle_flags(internal::evaluator<Dst>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010053 traits::debug();
Gael Guennebaud51ec1882010-07-08 23:30:16 +020054 std::cerr << " Expected Traversal == " << demangle_traversal(traversal)
Gael Guennebauda3950242014-03-12 18:14:58 +010055 << " got " << demangle_traversal(traits::Traversal) << "\n";
Gael Guennebaud51ec1882010-07-08 23:30:16 +020056 std::cerr << " Expected Unrolling == " << demangle_unrolling(unrolling)
Gael Guennebauda3950242014-03-12 18:14:58 +010057 << " got " << demangle_unrolling(traits::Unrolling) << "\n";
Gael Guennebaud51ec1882010-07-08 23:30:16 +020058 }
59 return res;
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000060}
61
Benoit Jacobb1f666d2010-02-26 20:12:51 -050062template<typename Dst, typename Src>
63bool test_assign(int traversal, int unrolling)
64{
Christoph Hertzberge3c82892018-09-21 21:15:51 +020065 EIGEN_STATIC_ASSERT_SAME_MATRIX_SIZE(Dst,Src);
Gael Guennebaud66e99ab2016-06-06 15:11:41 +020066 typedef internal::copy_using_evaluator_traits<internal::evaluator<Dst>,internal::evaluator<Src>, internal::assign_op<typename Dst::Scalar,typename Src::Scalar> > traits;
Gael Guennebauda3950242014-03-12 18:14:58 +010067 bool res = traits::Traversal==traversal && traits::Unrolling==unrolling;
Gael Guennebaud51ec1882010-07-08 23:30:16 +020068 if(!res)
69 {
Gael Guennebauda3950242014-03-12 18:14:58 +010070 std::cerr << "Src: " << demangle_flags(Src::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010071 std::cerr << " " << demangle_flags(internal::evaluator<Src>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010072 std::cerr << "Dst: " << demangle_flags(Dst::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010073 std::cerr << " " << demangle_flags(internal::evaluator<Dst>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010074 traits::debug();
Gael Guennebaud51ec1882010-07-08 23:30:16 +020075 std::cerr << " Expected Traversal == " << demangle_traversal(traversal)
Gael Guennebauda3950242014-03-12 18:14:58 +010076 << " got " << demangle_traversal(traits::Traversal) << "\n";
Gael Guennebaud51ec1882010-07-08 23:30:16 +020077 std::cerr << " Expected Unrolling == " << demangle_unrolling(unrolling)
Gael Guennebauda3950242014-03-12 18:14:58 +010078 << " got " << demangle_unrolling(traits::Unrolling) << "\n";
Gael Guennebaud51ec1882010-07-08 23:30:16 +020079 }
80 return res;
Benoit Jacobb1f666d2010-02-26 20:12:51 -050081}
82
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000083template<typename Xpr>
Benoit Jacob94c706d2009-11-18 11:57:07 -050084bool test_redux(const Xpr&, int traversal, int unrolling)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000085{
Gael Guennebaud66e99ab2016-06-06 15:11:41 +020086 typedef typename Xpr::Scalar Scalar;
87 typedef internal::redux_traits<internal::scalar_sum_op<Scalar,Scalar>,internal::redux_evaluator<Xpr> > traits;
Gael Guennebauda3950242014-03-12 18:14:58 +010088
Gael Guennebaud51ec1882010-07-08 23:30:16 +020089 bool res = traits::Traversal==traversal && traits::Unrolling==unrolling;
90 if(!res)
91 {
Gael Guennebauda3950242014-03-12 18:14:58 +010092 std::cerr << demangle_flags(Xpr::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010093 std::cerr << demangle_flags(internal::evaluator<Xpr>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010094 traits::debug();
95
Gael Guennebaud51ec1882010-07-08 23:30:16 +020096 std::cerr << " Expected Traversal == " << demangle_traversal(traversal)
97 << " got " << demangle_traversal(traits::Traversal) << "\n";
98 std::cerr << " Expected Unrolling == " << demangle_unrolling(unrolling)
99 << " got " << demangle_unrolling(traits::Unrolling) << "\n";
100 }
101 return res;
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000102}
103
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200104template<typename Scalar, bool Enable = internal::packet_traits<Scalar>::Vectorizable>
105struct vectorization_logic
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200106{
Christoph Hertzberg4f440b82014-07-14 14:36:20 +0200107 typedef internal::packet_traits<Scalar> PacketTraits;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200108
109 typedef typename internal::packet_traits<Scalar>::type PacketType;
110 typedef typename internal::unpacket_traits<PacketType>::half HalfPacketType;
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200111 enum {
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200112 PacketSize = internal::unpacket_traits<PacketType>::size,
113 HalfPacketSize = internal::unpacket_traits<HalfPacketType>::size
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200114 };
115 static void run()
116 {
117
118 typedef Matrix<Scalar,PacketSize,1> Vector1;
119 typedef Matrix<Scalar,Dynamic,1> VectorX;
120 typedef Matrix<Scalar,Dynamic,Dynamic> MatrixXX;
121 typedef Matrix<Scalar,PacketSize,PacketSize> Matrix11;
Gael Guennebaud91716f02018-09-21 14:32:24 +0200122 typedef Matrix<Scalar,(Matrix11::Flags&RowMajorBit)?8:2*PacketSize,(Matrix11::Flags&RowMajorBit)?2*PacketSize:8> Matrix22;
Gael Guennebaud64356a62011-01-04 14:18:07 +0100123 typedef Matrix<Scalar,(Matrix11::Flags&RowMajorBit)?16:4*PacketSize,(Matrix11::Flags&RowMajorBit)?4*PacketSize:16> Matrix44;
124 typedef Matrix<Scalar,(Matrix11::Flags&RowMajorBit)?16:4*PacketSize,(Matrix11::Flags&RowMajorBit)?4*PacketSize:16,DontAlign|EIGEN_DEFAULT_MATRIX_STORAGE_ORDER_OPTION> Matrix44u;
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200125 typedef Matrix<Scalar,4*PacketSize,4*PacketSize,ColMajor> Matrix44c;
126 typedef Matrix<Scalar,4*PacketSize,4*PacketSize,RowMajor> Matrix44r;
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200127
128 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200129 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
130 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200131 > Matrix1;
132
133 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200134 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
135 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1),
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200136 DontAlign|((Matrix1::Flags&RowMajorBit)?RowMajor:ColMajor)> Matrix1u;
137
Gael Guennebaud64356a62011-01-04 14:18:07 +0100138 // this type is made such that it can only be vectorized when viewed as a linear 1D vector
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200139 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200140 (PacketSize==16 ? 4 : PacketSize==8 ? 4 : PacketSize==4 ? 6 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?2:3) : /*PacketSize==1 ?*/ 1),
141 (PacketSize==16 ? 12 : PacketSize==8 ? 6 : PacketSize==4 ? 2 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?3:2) : /*PacketSize==1 ?*/ 3)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200142 > Matrix3;
Gael Guennebaud501bc602011-05-19 21:52:40 +0200143
144 #if !EIGEN_GCC_AND_ARCH_DOESNT_WANT_STACK_ALIGNMENT
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200145 VERIFY(test_assign(Vector1(),Vector1(),
146 InnerVectorizedTraversal,CompleteUnrolling));
147 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
148 InnerVectorizedTraversal,CompleteUnrolling));
149 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
150 InnerVectorizedTraversal,CompleteUnrolling));
151 VERIFY(test_assign(Vector1(),Vector1().template cast<Scalar>(),
152 InnerVectorizedTraversal,CompleteUnrolling));
153
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200154 VERIFY(test_assign(Matrix44(),Matrix44()+Matrix44(),
155 InnerVectorizedTraversal,InnerUnrolling));
156
157 VERIFY(test_assign(Matrix44u(),Matrix44()+Matrix44(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200158 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearTraversal,
159 EIGEN_UNALIGNED_VECTORIZE ? InnerUnrolling : NoUnrolling));
160
161 VERIFY(test_assign(Matrix1(),Matrix1()+Matrix1(),
Rasmus Munk Larsenf64b2952021-06-10 17:17:39 -0700162 (int(Matrix1::InnerSizeAtCompileTime) % int(PacketSize))==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal,
Gael Guennebaude68e1652016-05-24 21:54:03 +0200163 CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200164
165 VERIFY(test_assign(Matrix1u(),Matrix1()+Matrix1(),
Rasmus Munk Larsenf64b2952021-06-10 17:17:39 -0700166 EIGEN_UNALIGNED_VECTORIZE ? ((int(Matrix1::InnerSizeAtCompileTime) % int(PacketSize))==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal)
Gael Guennebaude68e1652016-05-24 21:54:03 +0200167 : LinearTraversal, CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200168
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200169 VERIFY(test_assign(Matrix44c().col(1),Matrix44c().col(2)+Matrix44c().col(3),
170 InnerVectorizedTraversal,CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700171
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200172 VERIFY(test_assign(Matrix44r().row(2),Matrix44r().row(1)+Matrix44r().row(1),
173 InnerVectorizedTraversal,CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700174
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200175 if(PacketSize>1)
176 {
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200177 typedef Matrix<Scalar,3,3,ColMajor> Matrix33c;
Gael Guennebaude68e1652016-05-24 21:54:03 +0200178 typedef Matrix<Scalar,3,1,ColMajor> Vector3;
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200179 VERIFY(test_assign(Matrix33c().row(2),Matrix33c().row(1)+Matrix33c().row(1),
180 LinearTraversal,CompleteUnrolling));
Gael Guennebaude68e1652016-05-24 21:54:03 +0200181 VERIFY(test_assign(Vector3(),Vector3()+Vector3(),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100182 sizeof(Scalar)==16 ? InnerVectorizedTraversal : (EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal), CompleteUnrolling));
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200183 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100184 EIGEN_UNALIGNED_VECTORIZE ? (sizeof(Scalar)==16 ? InnerVectorizedTraversal : LinearVectorizedTraversal)
185 : (sizeof(Scalar)==16 ? SliceVectorizedTraversal : LinearTraversal),
186 ((!EIGEN_UNALIGNED_VECTORIZE) && (sizeof(Scalar)==16)) ? NoUnrolling : CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700187
188 VERIFY(test_assign(Matrix3(),Matrix3().cwiseProduct(Matrix3()),
189 LinearVectorizedTraversal,CompleteUnrolling));
190
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200191 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100192 sizeof(Scalar)==16 ? InnerVectorizedTraversal :
Gael Guennebaude68e1652016-05-24 21:54:03 +0200193 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal :
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100194 LinearTraversal,
Gael Guennebaude68e1652016-05-24 21:54:03 +0200195 NoUnrolling));
196
197 VERIFY(test_assign(Matrix11(), Matrix11()+Matrix11(),InnerVectorizedTraversal,CompleteUnrolling));
198
Benoit Steiner25d05c42016-04-12 14:13:25 -0700199
Gael Guennebaud91716f02018-09-21 14:32:24 +0200200 VERIFY(test_assign(Matrix11(),Matrix<Scalar,21,21>().template block<PacketSize,PacketSize>(2,3)+Matrix<Scalar,21,21>().template block<PacketSize,PacketSize>(3,2),
Gael Guennebaud5ca24572016-07-06 22:25:24 +0200201 (EIGEN_UNALIGNED_VECTORIZE) ? InnerVectorizedTraversal : DefaultTraversal, CompleteUnrolling|InnerUnrolling));
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200202
203 VERIFY(test_assign(Vector1(),Matrix11()*Vector1(),
204 InnerVectorizedTraversal,CompleteUnrolling));
205
206 VERIFY(test_assign(Matrix11(),Matrix11().lazyProduct(Matrix11()),
207 InnerVectorizedTraversal,InnerUnrolling+CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200208 }
Benoit Steiner25d05c42016-04-12 14:13:25 -0700209
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200210 VERIFY(test_redux(Vector1(),
211 LinearVectorizedTraversal,CompleteUnrolling));
212
Gael Guennebaud9c3aed92017-12-14 14:24:33 +0100213 VERIFY(test_redux(Vector1().array()*Vector1().array(),
214 LinearVectorizedTraversal,CompleteUnrolling));
215
216 VERIFY(test_redux((Vector1().array()*Vector1().array()).col(0),
217 LinearVectorizedTraversal,CompleteUnrolling));
218
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200219 VERIFY(test_redux(Matrix<Scalar,PacketSize,3>(),
220 LinearVectorizedTraversal,CompleteUnrolling));
221
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200222 VERIFY(test_redux(Matrix3(),
223 LinearVectorizedTraversal,CompleteUnrolling));
224
225 VERIFY(test_redux(Matrix44(),
226 LinearVectorizedTraversal,NoUnrolling));
227
Gael Guennebaudb00e48a2018-09-21 13:45:56 +0200228 if(PacketSize>1) {
229 VERIFY(test_redux(Matrix44().template block<(Matrix1::Flags&RowMajorBit)?4:PacketSize,(Matrix1::Flags&RowMajorBit)?PacketSize:4>(1,2),
230 SliceVectorizedTraversal,CompleteUnrolling));
231
232 VERIFY(test_redux(Matrix44().template block<(Matrix1::Flags&RowMajorBit)?2:PacketSize,(Matrix1::Flags&RowMajorBit)?PacketSize:2>(1,2),
233 DefaultTraversal,CompleteUnrolling));
234 }
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200235
236 VERIFY(test_redux(Matrix44c().template block<2*PacketSize,1>(1,2),
237 LinearVectorizedTraversal,CompleteUnrolling));
238
239 VERIFY(test_redux(Matrix44r().template block<1,2*PacketSize>(2,1),
240 LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200241
Gael Guennebaud501bc602011-05-19 21:52:40 +0200242 VERIFY((test_assign<
Gael Guennebaud1f502432015-08-06 15:31:07 +0200243 Map<Matrix22, AlignedMax, OuterStride<3*PacketSize> >,
Gael Guennebaud501bc602011-05-19 21:52:40 +0200244 Matrix22
245 >(InnerVectorizedTraversal,CompleteUnrolling)));
246
247 VERIFY((test_assign<
Gael Guennebaud1f502432015-08-06 15:31:07 +0200248 Map<Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>, AlignedMax, InnerStride<3*PacketSize> >,
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200249 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
Gael Guennebaud5ca24572016-07-06 22:25:24 +0200250 >(DefaultTraversal,PacketSize>=8?InnerUnrolling:CompleteUnrolling)));
Gael Guennebaud501bc602011-05-19 21:52:40 +0200251
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200252 VERIFY((test_assign(Matrix11(), Matrix<Scalar,PacketSize,EIGEN_PLAIN_ENUM_MIN(2,PacketSize)>()*Matrix<Scalar,EIGEN_PLAIN_ENUM_MIN(2,PacketSize),PacketSize>(),
Gael Guennebaud1330f8b2015-03-13 21:15:50 +0100253 InnerVectorizedTraversal, CompleteUnrolling)));
Gael Guennebaud501bc602011-05-19 21:52:40 +0200254 #endif
255
256 VERIFY(test_assign(MatrixXX(10,10),MatrixXX(20,20).block(10,10,2,3),
257 SliceVectorizedTraversal,NoUnrolling));
258
259 VERIFY(test_redux(VectorX(10),
260 LinearVectorizedTraversal,NoUnrolling));
Francesco Mazzolieb85b972021-11-29 21:13:24 +0000261
262 // Some static checks for packet-picking -- see
263 // <https://gitlab.com/libeigen/eigen/merge_requests/46#note_271497656> for context.
264
265 // Any multiple of the packet size itself will result in the normal packet
266 STATIC_CHECK((
267 internal::is_same<typename internal::find_best_packet<Scalar, PacketSize>::type, PacketType>::value
268 ));
269 STATIC_CHECK((
270 internal::is_same<typename internal::find_best_packet<Scalar, PacketSize*2>::type, PacketType>::value
271 ));
272 STATIC_CHECK((
273 internal::is_same<typename internal::find_best_packet<Scalar, PacketSize*5>::type, PacketType>::value
274 ));
275 // Moreover, situations where the size is _not_ a multiple but picking the full packet
276 // is convenient will also work, but only with unaligned vectorize
277 STATIC_CHECK((
278 !(EIGEN_UNALIGNED_VECTORIZE || PacketSize == HalfPacketSize) ||
279 internal::is_same<typename internal::find_best_packet<Scalar, PacketSize*5+1>::type, PacketType>::value
280 ));
281 STATIC_CHECK((
282 !(EIGEN_UNALIGNED_VECTORIZE || PacketSize == HalfPacketSize) ||
283 internal::is_same<typename internal::find_best_packet<Scalar, PacketSize*5+2>::type, PacketType>::value
284 ));
285 // In situations where the picking the full-packet would be detrimental the half-packet
286 // is chosen.
287 STATIC_CHECK((
288 !(PacketSize > 2) ||
289 internal::is_same<typename internal::find_best_packet<Scalar, PacketSize*2-1>::type, HalfPacketType>::value
290 ));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200291 }
Francesco Mazzolieb85b972021-11-29 21:13:24 +0000292
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200293};
294
295template<typename Scalar> struct vectorization_logic<Scalar,false>
296{
297 static void run() {}
298};
299
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200300template<typename Scalar, bool Enable = !internal::is_same<typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half,
301 typename internal::packet_traits<Scalar>::type>::value >
302struct vectorization_logic_half
303{
304 typedef internal::packet_traits<Scalar> PacketTraits;
305 typedef typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half PacketType;
306 enum {
307 PacketSize = internal::unpacket_traits<PacketType>::size
308 };
309 static void run()
310 {
311
312 typedef Matrix<Scalar,PacketSize,1> Vector1;
313 typedef Matrix<Scalar,PacketSize,PacketSize> Matrix11;
314 typedef Matrix<Scalar,5*PacketSize,7,ColMajor> Matrix57;
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200315 typedef Matrix<Scalar,3*PacketSize,5,ColMajor> Matrix35;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200316 typedef Matrix<Scalar,5*PacketSize,7,DontAlign|ColMajor> Matrix57u;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200317
318 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200319 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
320 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1)
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200321 > Matrix1;
322
323 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200324 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
325 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1),
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200326 DontAlign|((Matrix1::Flags&RowMajorBit)?RowMajor:ColMajor)> Matrix1u;
327
328 // this type is made such that it can only be vectorized when viewed as a linear 1D vector
329 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200330 (PacketSize==16 ? 4 : PacketSize==8 ? 4 : PacketSize==4 ? 6 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?2:3) : /*PacketSize==1 ?*/ 1),
331 (PacketSize==16 ? 12 : PacketSize==8 ? 6 : PacketSize==4 ? 2 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?3:2) : /*PacketSize==1 ?*/ 3)
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200332 > Matrix3;
333
334 #if !EIGEN_GCC_AND_ARCH_DOESNT_WANT_STACK_ALIGNMENT
335 VERIFY(test_assign(Vector1(),Vector1(),
336 InnerVectorizedTraversal,CompleteUnrolling));
337 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
338 InnerVectorizedTraversal,CompleteUnrolling));
Gael Guennebaude68e1652016-05-24 21:54:03 +0200339 VERIFY(test_assign(Vector1(),Vector1().template segment<PacketSize>(0).derived(),
340 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
341 VERIFY(test_assign(Vector1(),Scalar(2.1)*Vector1()-Vector1(),
342 InnerVectorizedTraversal,CompleteUnrolling));
343 VERIFY(test_assign(Vector1(),(Scalar(2.1)*Vector1().template segment<PacketSize>(0)-Vector1().template segment<PacketSize>(0)).derived(),
344 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200345 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
346 InnerVectorizedTraversal,CompleteUnrolling));
347 VERIFY(test_assign(Vector1(),Vector1().template cast<Scalar>(),
348 InnerVectorizedTraversal,CompleteUnrolling));
349
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200350 VERIFY(test_assign(Matrix57(),Matrix57()+Matrix57(),
351 InnerVectorizedTraversal,InnerUnrolling));
352
353 VERIFY(test_assign(Matrix57u(),Matrix57()+Matrix57(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200354 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearTraversal,
355 EIGEN_UNALIGNED_VECTORIZE ? InnerUnrolling : NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200356
357 VERIFY(test_assign(Matrix1u(),Matrix1()+Matrix1(),
Rasmus Munk Larsen13fb5ab2021-06-15 09:09:31 -0700358 EIGEN_UNALIGNED_VECTORIZE ? ((int(Matrix1::InnerSizeAtCompileTime) % int(PacketSize))==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200359
360 if(PacketSize>1)
361 {
362 typedef Matrix<Scalar,3,3,ColMajor> Matrix33c;
363 VERIFY(test_assign(Matrix33c().row(2),Matrix33c().row(1)+Matrix33c().row(1),
364 LinearTraversal,CompleteUnrolling));
365 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100366 EIGEN_UNALIGNED_VECTORIZE ? (sizeof(Scalar)==16 ? InnerVectorizedTraversal : LinearVectorizedTraversal)
367 : (sizeof(Scalar)==16 ? SliceVectorizedTraversal : LinearTraversal),
368 ((!EIGEN_UNALIGNED_VECTORIZE) && (sizeof(Scalar)==16)) ? NoUnrolling : CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200369
370 VERIFY(test_assign(Matrix3(),Matrix3().cwiseQuotient(Matrix3()),
Jakub Lichmandc5b1f72021-08-25 19:38:23 +0000371 PacketTraits::HasDiv ? LinearVectorizedTraversal : LinearTraversal,
372 PacketTraits::HasDiv ? CompleteUnrolling : NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200373
374 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100375 sizeof(Scalar)==16 ? InnerVectorizedTraversal : (EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200376 NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200377
378 VERIFY(test_assign(Matrix11(),Matrix<Scalar,17,17>().template block<PacketSize,PacketSize>(2,3)+Matrix<Scalar,17,17>().template block<PacketSize,PacketSize>(8,4),
Gael Guennebaud91716f02018-09-21 14:32:24 +0200379 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : DefaultTraversal,InnerUnrolling+CompleteUnrolling));
380
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200381
382 VERIFY(test_assign(Vector1(),Matrix11()*Vector1(),
383 InnerVectorizedTraversal,CompleteUnrolling));
384
385 VERIFY(test_assign(Matrix11(),Matrix11().lazyProduct(Matrix11()),
386 InnerVectorizedTraversal,InnerUnrolling+CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200387 }
388
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200389 VERIFY(test_redux(Vector1(),
390 LinearVectorizedTraversal,CompleteUnrolling));
391
392 VERIFY(test_redux(Matrix<Scalar,PacketSize,3>(),
393 LinearVectorizedTraversal,CompleteUnrolling));
394
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200395 VERIFY(test_redux(Matrix3(),
396 LinearVectorizedTraversal,CompleteUnrolling));
397
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200398 VERIFY(test_redux(Matrix35(),
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200399 LinearVectorizedTraversal,CompleteUnrolling));
400
Gael Guennebaudb00e48a2018-09-21 13:45:56 +0200401 VERIFY(test_redux(Matrix57().template block<PacketSize==1?2:PacketSize,3>(1,0),
402 SliceVectorizedTraversal,CompleteUnrolling));
403
404 if(PacketSize>1) {
405 VERIFY(test_redux(Matrix57().template block<PacketSize,2>(1,0),
406 DefaultTraversal,CompleteUnrolling));
407 }
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200408
409 VERIFY((test_assign<
410 Map<Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>, AlignedMax, InnerStride<3*PacketSize> >,
411 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
Gael Guennebaud91716f02018-09-21 14:32:24 +0200412 >(DefaultTraversal,PacketSize>4?InnerUnrolling:CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200413
414 VERIFY((test_assign(Matrix57(), Matrix<Scalar,5*PacketSize,3>()*Matrix<Scalar,3,7>(),
Gael Guennebaud91716f02018-09-21 14:32:24 +0200415 InnerVectorizedTraversal, InnerUnrolling+CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200416 #endif
417 }
418};
419
420template<typename Scalar> struct vectorization_logic_half<Scalar,false>
421{
422 static void run() {}
423};
424
Gael Guennebaud82f0ce22018-07-17 14:46:15 +0200425EIGEN_DECLARE_TEST(vectorization_logic)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000426{
427
428#ifdef EIGEN_VECTORIZE
429
Christoph Hertzberg4f440b82014-07-14 14:36:20 +0200430 CALL_SUBTEST( vectorization_logic<int>::run() );
Gael Guennebaud501bc602011-05-19 21:52:40 +0200431 CALL_SUBTEST( vectorization_logic<float>::run() );
432 CALL_SUBTEST( vectorization_logic<double>::run() );
433 CALL_SUBTEST( vectorization_logic<std::complex<float> >::run() );
434 CALL_SUBTEST( vectorization_logic<std::complex<double> >::run() );
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200435
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200436 CALL_SUBTEST( vectorization_logic_half<int>::run() );
437 CALL_SUBTEST( vectorization_logic_half<float>::run() );
438 CALL_SUBTEST( vectorization_logic_half<double>::run() );
439 CALL_SUBTEST( vectorization_logic_half<std::complex<float> >::run() );
440 CALL_SUBTEST( vectorization_logic_half<std::complex<double> >::run() );
441
Benoit Jacob47160402010-10-25 10:15:22 -0400442 if(internal::packet_traits<float>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200443 {
444 VERIFY(test_assign(Matrix<float,3,3>(),Matrix<float,3,3>()+Matrix<float,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200445 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200446
447 VERIFY(test_redux(Matrix<float,5,2>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200448 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200449 }
450
Benoit Jacob47160402010-10-25 10:15:22 -0400451 if(internal::packet_traits<double>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200452 {
453 VERIFY(test_assign(Matrix<double,3,3>(),Matrix<double,3,3>()+Matrix<double,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200454 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200455
456 VERIFY(test_redux(Matrix<double,7,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200457 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200458 }
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000459#endif // EIGEN_VECTORIZE
460
461}