blob: 01b55e192d698f935e9ed18d521f88bf71de6c90 [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 Guennebaud0fc89542015-10-27 10:38:49 +010025using internal::demangle_flags;
26using internal::demangle_traversal;
27using internal::demangle_unrolling;
Gael Guennebaud51ec1882010-07-08 23:30:16 +020028
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000029template<typename Dst, typename Src>
Benoit Jacob94c706d2009-11-18 11:57:07 -050030bool test_assign(const Dst&, const Src&, int traversal, int unrolling)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000031{
Gael Guennebaud66e99ab2016-06-06 15:11:41 +020032 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 +020033 bool res = traits::Traversal==traversal;
34 if(unrolling==InnerUnrolling+CompleteUnrolling)
Gael Guennebauda95e1a22016-04-13 22:00:38 +020035 res = res && (int(traits::Unrolling)==InnerUnrolling || int(traits::Unrolling)==CompleteUnrolling);
Gael Guennebaud512ba0a2016-04-13 18:16:35 +020036 else
Gael Guennebauda95e1a22016-04-13 22:00:38 +020037 res = res && int(traits::Unrolling)==unrolling;
Gael Guennebaud51ec1882010-07-08 23:30:16 +020038 if(!res)
39 {
Gael Guennebauda3950242014-03-12 18:14:58 +010040 std::cerr << "Src: " << demangle_flags(Src::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010041 std::cerr << " " << demangle_flags(internal::evaluator<Src>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010042 std::cerr << "Dst: " << demangle_flags(Dst::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010043 std::cerr << " " << demangle_flags(internal::evaluator<Dst>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010044 traits::debug();
Gael Guennebaud51ec1882010-07-08 23:30:16 +020045 std::cerr << " Expected Traversal == " << demangle_traversal(traversal)
Gael Guennebauda3950242014-03-12 18:14:58 +010046 << " got " << demangle_traversal(traits::Traversal) << "\n";
Gael Guennebaud51ec1882010-07-08 23:30:16 +020047 std::cerr << " Expected Unrolling == " << demangle_unrolling(unrolling)
Gael Guennebauda3950242014-03-12 18:14:58 +010048 << " got " << demangle_unrolling(traits::Unrolling) << "\n";
Gael Guennebaud51ec1882010-07-08 23:30:16 +020049 }
50 return res;
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000051}
52
Benoit Jacobb1f666d2010-02-26 20:12:51 -050053template<typename Dst, typename Src>
54bool test_assign(int traversal, int unrolling)
55{
Gael Guennebaud66e99ab2016-06-06 15:11:41 +020056 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 +010057 bool res = traits::Traversal==traversal && traits::Unrolling==unrolling;
Gael Guennebaud51ec1882010-07-08 23:30:16 +020058 if(!res)
59 {
Gael Guennebauda3950242014-03-12 18:14:58 +010060 std::cerr << "Src: " << demangle_flags(Src::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010061 std::cerr << " " << demangle_flags(internal::evaluator<Src>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010062 std::cerr << "Dst: " << demangle_flags(Dst::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010063 std::cerr << " " << demangle_flags(internal::evaluator<Dst>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010064 traits::debug();
Gael Guennebaud51ec1882010-07-08 23:30:16 +020065 std::cerr << " Expected Traversal == " << demangle_traversal(traversal)
Gael Guennebauda3950242014-03-12 18:14:58 +010066 << " got " << demangle_traversal(traits::Traversal) << "\n";
Gael Guennebaud51ec1882010-07-08 23:30:16 +020067 std::cerr << " Expected Unrolling == " << demangle_unrolling(unrolling)
Gael Guennebauda3950242014-03-12 18:14:58 +010068 << " got " << demangle_unrolling(traits::Unrolling) << "\n";
Gael Guennebaud51ec1882010-07-08 23:30:16 +020069 }
70 return res;
Benoit Jacobb1f666d2010-02-26 20:12:51 -050071}
72
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000073template<typename Xpr>
Benoit Jacob94c706d2009-11-18 11:57:07 -050074bool test_redux(const Xpr&, int traversal, int unrolling)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000075{
Gael Guennebaud66e99ab2016-06-06 15:11:41 +020076 typedef typename Xpr::Scalar Scalar;
77 typedef internal::redux_traits<internal::scalar_sum_op<Scalar,Scalar>,internal::redux_evaluator<Xpr> > traits;
Gael Guennebauda3950242014-03-12 18:14:58 +010078
Gael Guennebaud51ec1882010-07-08 23:30:16 +020079 bool res = traits::Traversal==traversal && traits::Unrolling==unrolling;
80 if(!res)
81 {
Gael Guennebauda3950242014-03-12 18:14:58 +010082 std::cerr << demangle_flags(Xpr::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010083 std::cerr << demangle_flags(internal::evaluator<Xpr>::Flags) << std::endl;
Gael Guennebauda3950242014-03-12 18:14:58 +010084 traits::debug();
85
Gael Guennebaud51ec1882010-07-08 23:30:16 +020086 std::cerr << " Expected Traversal == " << demangle_traversal(traversal)
87 << " got " << demangle_traversal(traits::Traversal) << "\n";
88 std::cerr << " Expected Unrolling == " << demangle_unrolling(unrolling)
89 << " got " << demangle_unrolling(traits::Unrolling) << "\n";
90 }
91 return res;
Gael Guennebaudcb71dc42009-01-07 22:20:03 +000092}
93
Gael Guennebaudfebcce32015-08-07 20:05:31 +020094template<typename Scalar, bool Enable = internal::packet_traits<Scalar>::Vectorizable>
95struct vectorization_logic
Gael Guennebaud51ec1882010-07-08 23:30:16 +020096{
Christoph Hertzberg4f440b82014-07-14 14:36:20 +020097 typedef internal::packet_traits<Scalar> PacketTraits;
Gael Guennebaudfebcce32015-08-07 20:05:31 +020098
99 typedef typename internal::packet_traits<Scalar>::type PacketType;
100 typedef typename internal::unpacket_traits<PacketType>::half HalfPacketType;
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200101 enum {
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200102 PacketSize = internal::unpacket_traits<PacketType>::size,
103 HalfPacketSize = internal::unpacket_traits<HalfPacketType>::size
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200104 };
105 static void run()
106 {
107
108 typedef Matrix<Scalar,PacketSize,1> Vector1;
109 typedef Matrix<Scalar,Dynamic,1> VectorX;
110 typedef Matrix<Scalar,Dynamic,Dynamic> MatrixXX;
111 typedef Matrix<Scalar,PacketSize,PacketSize> Matrix11;
112 typedef Matrix<Scalar,2*PacketSize,2*PacketSize> Matrix22;
Gael Guennebaud64356a62011-01-04 14:18:07 +0100113 typedef Matrix<Scalar,(Matrix11::Flags&RowMajorBit)?16:4*PacketSize,(Matrix11::Flags&RowMajorBit)?4*PacketSize:16> Matrix44;
114 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 +0200115 typedef Matrix<Scalar,4*PacketSize,4*PacketSize,ColMajor> Matrix44c;
116 typedef Matrix<Scalar,4*PacketSize,4*PacketSize,RowMajor> Matrix44r;
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200117
118 typedef Matrix<Scalar,
119 (PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
120 (PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1)
121 > Matrix1;
122
123 typedef Matrix<Scalar,
124 (PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
125 (PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1),
126 DontAlign|((Matrix1::Flags&RowMajorBit)?RowMajor:ColMajor)> Matrix1u;
127
Gael Guennebaud64356a62011-01-04 14:18:07 +0100128 // 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 +0200129 typedef Matrix<Scalar,
Gael Guennebaud64356a62011-01-04 14:18:07 +0100130 (PacketSize==8 ? 4 : PacketSize==4 ? 6 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?2:3) : /*PacketSize==1 ?*/ 1),
131 (PacketSize==8 ? 6 : PacketSize==4 ? 2 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?3:2) : /*PacketSize==1 ?*/ 3)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200132 > Matrix3;
Gael Guennebaud501bc602011-05-19 21:52:40 +0200133
134 #if !EIGEN_GCC_AND_ARCH_DOESNT_WANT_STACK_ALIGNMENT
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200135 VERIFY(test_assign(Vector1(),Vector1(),
136 InnerVectorizedTraversal,CompleteUnrolling));
137 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
138 InnerVectorizedTraversal,CompleteUnrolling));
139 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
140 InnerVectorizedTraversal,CompleteUnrolling));
141 VERIFY(test_assign(Vector1(),Vector1().template cast<Scalar>(),
142 InnerVectorizedTraversal,CompleteUnrolling));
143
144
145 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
152 VERIFY(test_assign(Matrix44(),Matrix44()+Matrix44(),
153 InnerVectorizedTraversal,InnerUnrolling));
154
155 VERIFY(test_assign(Matrix44u(),Matrix44()+Matrix44(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200156 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearTraversal,
157 EIGEN_UNALIGNED_VECTORIZE ? InnerUnrolling : NoUnrolling));
158
159 VERIFY(test_assign(Matrix1(),Matrix1()+Matrix1(),
160 (Matrix1::InnerSizeAtCompileTime % PacketSize)==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal,
161 CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200162
163 VERIFY(test_assign(Matrix1u(),Matrix1()+Matrix1(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200164 EIGEN_UNALIGNED_VECTORIZE ? ((Matrix1::InnerSizeAtCompileTime % PacketSize)==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal)
165 : LinearTraversal, CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200166
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200167 VERIFY(test_assign(Matrix44c().col(1),Matrix44c().col(2)+Matrix44c().col(3),
168 InnerVectorizedTraversal,CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700169
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200170 VERIFY(test_assign(Matrix44r().row(2),Matrix44r().row(1)+Matrix44r().row(1),
171 InnerVectorizedTraversal,CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700172
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200173 if(PacketSize>1)
174 {
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200175 typedef Matrix<Scalar,3,3,ColMajor> Matrix33c;
Gael Guennebaude68e1652016-05-24 21:54:03 +0200176 typedef Matrix<Scalar,3,1,ColMajor> Vector3;
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200177 VERIFY(test_assign(Matrix33c().row(2),Matrix33c().row(1)+Matrix33c().row(1),
178 LinearTraversal,CompleteUnrolling));
Gael Guennebaude68e1652016-05-24 21:54:03 +0200179 VERIFY(test_assign(Vector3(),Vector3()+Vector3(),
180 EIGEN_UNALIGNED_VECTORIZE ? (HalfPacketSize==1 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : (HalfPacketSize==1 ? InnerVectorizedTraversal : LinearTraversal), CompleteUnrolling));
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200181 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200182 EIGEN_UNALIGNED_VECTORIZE ? (HalfPacketSize==1 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : (HalfPacketSize==1 ? SliceVectorizedTraversal : LinearTraversal),
183 ((!EIGEN_UNALIGNED_VECTORIZE) && HalfPacketSize==1) ? NoUnrolling : CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700184
185 VERIFY(test_assign(Matrix3(),Matrix3().cwiseProduct(Matrix3()),
186 LinearVectorizedTraversal,CompleteUnrolling));
187
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200188 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200189 HalfPacketSize==1 ? InnerVectorizedTraversal :
190 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal :
191 LinearTraversal,
192 NoUnrolling));
193
194 VERIFY(test_assign(Matrix11(), Matrix11()+Matrix11(),InnerVectorizedTraversal,CompleteUnrolling));
195
Benoit Steiner25d05c42016-04-12 14:13:25 -0700196
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200197 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 Guennebaud5ca24572016-07-06 22:25:24 +0200198 (EIGEN_UNALIGNED_VECTORIZE) ? InnerVectorizedTraversal : DefaultTraversal, CompleteUnrolling|InnerUnrolling));
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200199
200 VERIFY(test_assign(Vector1(),Matrix11()*Vector1(),
201 InnerVectorizedTraversal,CompleteUnrolling));
202
203 VERIFY(test_assign(Matrix11(),Matrix11().lazyProduct(Matrix11()),
204 InnerVectorizedTraversal,InnerUnrolling+CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200205 }
Benoit Steiner25d05c42016-04-12 14:13:25 -0700206
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200207 VERIFY(test_redux(Vector1(),
208 LinearVectorizedTraversal,CompleteUnrolling));
209
Gael Guennebaud9c3aed92017-12-14 14:24:33 +0100210 VERIFY(test_redux(Vector1().array()*Vector1().array(),
211 LinearVectorizedTraversal,CompleteUnrolling));
212
213 VERIFY(test_redux((Vector1().array()*Vector1().array()).col(0),
214 LinearVectorizedTraversal,CompleteUnrolling));
215
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200216 VERIFY(test_redux(Matrix<Scalar,PacketSize,3>(),
217 LinearVectorizedTraversal,CompleteUnrolling));
218
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200219 VERIFY(test_redux(Matrix3(),
220 LinearVectorizedTraversal,CompleteUnrolling));
221
222 VERIFY(test_redux(Matrix44(),
223 LinearVectorizedTraversal,NoUnrolling));
224
Gael Guennebaud64356a62011-01-04 14:18:07 +0100225 VERIFY(test_redux(Matrix44().template block<(Matrix1::Flags&RowMajorBit)?4:PacketSize,(Matrix1::Flags&RowMajorBit)?PacketSize:4>(1,2),
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200226 DefaultTraversal,CompleteUnrolling));
227
228 VERIFY(test_redux(Matrix44c().template block<2*PacketSize,1>(1,2),
229 LinearVectorizedTraversal,CompleteUnrolling));
230
231 VERIFY(test_redux(Matrix44r().template block<1,2*PacketSize>(2,1),
232 LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200233
Gael Guennebaud501bc602011-05-19 21:52:40 +0200234 VERIFY((test_assign<
Gael Guennebaud1f502432015-08-06 15:31:07 +0200235 Map<Matrix22, AlignedMax, OuterStride<3*PacketSize> >,
Gael Guennebaud501bc602011-05-19 21:52:40 +0200236 Matrix22
237 >(InnerVectorizedTraversal,CompleteUnrolling)));
238
239 VERIFY((test_assign<
Gael Guennebaud1f502432015-08-06 15:31:07 +0200240 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 +0200241 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
Gael Guennebaud5ca24572016-07-06 22:25:24 +0200242 >(DefaultTraversal,PacketSize>=8?InnerUnrolling:CompleteUnrolling)));
Gael Guennebaud501bc602011-05-19 21:52:40 +0200243
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200244 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 +0100245 InnerVectorizedTraversal, CompleteUnrolling)));
Gael Guennebaud501bc602011-05-19 21:52:40 +0200246 #endif
247
248 VERIFY(test_assign(MatrixXX(10,10),MatrixXX(20,20).block(10,10,2,3),
249 SliceVectorizedTraversal,NoUnrolling));
250
251 VERIFY(test_redux(VectorX(10),
252 LinearVectorizedTraversal,NoUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200253 }
254};
255
256template<typename Scalar> struct vectorization_logic<Scalar,false>
257{
258 static void run() {}
259};
260
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200261template<typename Scalar, bool Enable = !internal::is_same<typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half,
262 typename internal::packet_traits<Scalar>::type>::value >
263struct vectorization_logic_half
264{
265 typedef internal::packet_traits<Scalar> PacketTraits;
266 typedef typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half PacketType;
267 enum {
268 PacketSize = internal::unpacket_traits<PacketType>::size
269 };
270 static void run()
271 {
272
273 typedef Matrix<Scalar,PacketSize,1> Vector1;
274 typedef Matrix<Scalar,PacketSize,PacketSize> Matrix11;
275 typedef Matrix<Scalar,5*PacketSize,7,ColMajor> Matrix57;
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200276 typedef Matrix<Scalar,3*PacketSize,5,ColMajor> Matrix35;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200277 typedef Matrix<Scalar,5*PacketSize,7,DontAlign|ColMajor> Matrix57u;
278// typedef Matrix<Scalar,(Matrix11::Flags&RowMajorBit)?16:4*PacketSize,(Matrix11::Flags&RowMajorBit)?4*PacketSize:16> Matrix44;
279// typedef Matrix<Scalar,(Matrix11::Flags&RowMajorBit)?16:4*PacketSize,(Matrix11::Flags&RowMajorBit)?4*PacketSize:16,DontAlign|EIGEN_DEFAULT_MATRIX_STORAGE_ORDER_OPTION> Matrix44u;
280// typedef Matrix<Scalar,4*PacketSize,4*PacketSize,ColMajor> Matrix44c;
281// typedef Matrix<Scalar,4*PacketSize,4*PacketSize,RowMajor> Matrix44r;
282
283 typedef Matrix<Scalar,
284 (PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
285 (PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1)
286 > Matrix1;
287
288 typedef Matrix<Scalar,
289 (PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
290 (PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1),
291 DontAlign|((Matrix1::Flags&RowMajorBit)?RowMajor:ColMajor)> Matrix1u;
292
293 // this type is made such that it can only be vectorized when viewed as a linear 1D vector
294 typedef Matrix<Scalar,
295 (PacketSize==8 ? 4 : PacketSize==4 ? 6 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?2:3) : /*PacketSize==1 ?*/ 1),
296 (PacketSize==8 ? 6 : PacketSize==4 ? 2 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?3:2) : /*PacketSize==1 ?*/ 3)
297 > Matrix3;
298
299 #if !EIGEN_GCC_AND_ARCH_DOESNT_WANT_STACK_ALIGNMENT
300 VERIFY(test_assign(Vector1(),Vector1(),
301 InnerVectorizedTraversal,CompleteUnrolling));
302 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
303 InnerVectorizedTraversal,CompleteUnrolling));
Gael Guennebaude68e1652016-05-24 21:54:03 +0200304 VERIFY(test_assign(Vector1(),Vector1().template segment<PacketSize>(0).derived(),
305 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
306 VERIFY(test_assign(Vector1(),Scalar(2.1)*Vector1()-Vector1(),
307 InnerVectorizedTraversal,CompleteUnrolling));
308 VERIFY(test_assign(Vector1(),(Scalar(2.1)*Vector1().template segment<PacketSize>(0)-Vector1().template segment<PacketSize>(0)).derived(),
309 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200310 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
311 InnerVectorizedTraversal,CompleteUnrolling));
312 VERIFY(test_assign(Vector1(),Vector1().template cast<Scalar>(),
313 InnerVectorizedTraversal,CompleteUnrolling));
314
315
316 VERIFY(test_assign(Vector1(),Vector1(),
317 InnerVectorizedTraversal,CompleteUnrolling));
318 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
319 InnerVectorizedTraversal,CompleteUnrolling));
320 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
321 InnerVectorizedTraversal,CompleteUnrolling));
322
323 VERIFY(test_assign(Matrix57(),Matrix57()+Matrix57(),
324 InnerVectorizedTraversal,InnerUnrolling));
325
326 VERIFY(test_assign(Matrix57u(),Matrix57()+Matrix57(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200327 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearTraversal,
328 EIGEN_UNALIGNED_VECTORIZE ? InnerUnrolling : NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200329
330 VERIFY(test_assign(Matrix1u(),Matrix1()+Matrix1(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200331 EIGEN_UNALIGNED_VECTORIZE ? ((Matrix1::InnerSizeAtCompileTime % PacketSize)==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200332
333 if(PacketSize>1)
334 {
335 typedef Matrix<Scalar,3,3,ColMajor> Matrix33c;
336 VERIFY(test_assign(Matrix33c().row(2),Matrix33c().row(1)+Matrix33c().row(1),
337 LinearTraversal,CompleteUnrolling));
338 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200339 EIGEN_UNALIGNED_VECTORIZE ? (PacketSize==1 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200340
341 VERIFY(test_assign(Matrix3(),Matrix3().cwiseQuotient(Matrix3()),
342 PacketTraits::HasDiv ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
343
344 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200345 EIGEN_UNALIGNED_VECTORIZE ? (PacketSize==1 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,
346 NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200347
348 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 Guennebaude68e1652016-05-24 21:54:03 +0200349 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : DefaultTraversal,PacketSize>4?InnerUnrolling:CompleteUnrolling));
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200350
351 VERIFY(test_assign(Vector1(),Matrix11()*Vector1(),
352 InnerVectorizedTraversal,CompleteUnrolling));
353
354 VERIFY(test_assign(Matrix11(),Matrix11().lazyProduct(Matrix11()),
355 InnerVectorizedTraversal,InnerUnrolling+CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200356 }
357
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200358 VERIFY(test_redux(Vector1(),
359 LinearVectorizedTraversal,CompleteUnrolling));
360
361 VERIFY(test_redux(Matrix<Scalar,PacketSize,3>(),
362 LinearVectorizedTraversal,CompleteUnrolling));
363
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200364 VERIFY(test_redux(Matrix3(),
365 LinearVectorizedTraversal,CompleteUnrolling));
366
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200367 VERIFY(test_redux(Matrix35(),
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200368 LinearVectorizedTraversal,CompleteUnrolling));
369
370 VERIFY(test_redux(Matrix57().template block<PacketSize,3>(1,0),
371 DefaultTraversal,CompleteUnrolling));
372
373 VERIFY((test_assign<
374 Map<Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>, AlignedMax, InnerStride<3*PacketSize> >,
375 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
376 >(DefaultTraversal,CompleteUnrolling)));
377
378 VERIFY((test_assign(Matrix57(), Matrix<Scalar,5*PacketSize,3>()*Matrix<Scalar,3,7>(),
Gael Guennebaud5ca24572016-07-06 22:25:24 +0200379 InnerVectorizedTraversal, InnerUnrolling|CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200380 #endif
381 }
382};
383
384template<typename Scalar> struct vectorization_logic_half<Scalar,false>
385{
386 static void run() {}
387};
388
Gael Guennebaud82f0ce22018-07-17 14:46:15 +0200389EIGEN_DECLARE_TEST(vectorization_logic)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000390{
391
392#ifdef EIGEN_VECTORIZE
393
Christoph Hertzberg4f440b82014-07-14 14:36:20 +0200394 CALL_SUBTEST( vectorization_logic<int>::run() );
Gael Guennebaud501bc602011-05-19 21:52:40 +0200395 CALL_SUBTEST( vectorization_logic<float>::run() );
396 CALL_SUBTEST( vectorization_logic<double>::run() );
397 CALL_SUBTEST( vectorization_logic<std::complex<float> >::run() );
398 CALL_SUBTEST( vectorization_logic<std::complex<double> >::run() );
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200399
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200400 CALL_SUBTEST( vectorization_logic_half<int>::run() );
401 CALL_SUBTEST( vectorization_logic_half<float>::run() );
402 CALL_SUBTEST( vectorization_logic_half<double>::run() );
403 CALL_SUBTEST( vectorization_logic_half<std::complex<float> >::run() );
404 CALL_SUBTEST( vectorization_logic_half<std::complex<double> >::run() );
405
Benoit Jacob47160402010-10-25 10:15:22 -0400406 if(internal::packet_traits<float>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200407 {
408 VERIFY(test_assign(Matrix<float,3,3>(),Matrix<float,3,3>()+Matrix<float,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200409 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200410
411 VERIFY(test_redux(Matrix<float,5,2>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200412 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200413 }
414
Benoit Jacob47160402010-10-25 10:15:22 -0400415 if(internal::packet_traits<double>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200416 {
417 VERIFY(test_assign(Matrix<double,3,3>(),Matrix<double,3,3>()+Matrix<double,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200418 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200419
420 VERIFY(test_redux(Matrix<double,7,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200421 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200422 }
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000423#endif // EIGEN_VECTORIZE
424
425}