blob: 4bf3b3db2344a5caa8c2876420679cd240de58f1 [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
154
155 VERIFY(test_assign(Vector1(),Vector1(),
156 InnerVectorizedTraversal,CompleteUnrolling));
157 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
158 InnerVectorizedTraversal,CompleteUnrolling));
159 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
160 InnerVectorizedTraversal,CompleteUnrolling));
161
162 VERIFY(test_assign(Matrix44(),Matrix44()+Matrix44(),
163 InnerVectorizedTraversal,InnerUnrolling));
164
165 VERIFY(test_assign(Matrix44u(),Matrix44()+Matrix44(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200166 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearTraversal,
167 EIGEN_UNALIGNED_VECTORIZE ? InnerUnrolling : NoUnrolling));
168
169 VERIFY(test_assign(Matrix1(),Matrix1()+Matrix1(),
170 (Matrix1::InnerSizeAtCompileTime % PacketSize)==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal,
171 CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200172
173 VERIFY(test_assign(Matrix1u(),Matrix1()+Matrix1(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200174 EIGEN_UNALIGNED_VECTORIZE ? ((Matrix1::InnerSizeAtCompileTime % PacketSize)==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal)
175 : LinearTraversal, CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200176
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200177 VERIFY(test_assign(Matrix44c().col(1),Matrix44c().col(2)+Matrix44c().col(3),
178 InnerVectorizedTraversal,CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700179
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200180 VERIFY(test_assign(Matrix44r().row(2),Matrix44r().row(1)+Matrix44r().row(1),
181 InnerVectorizedTraversal,CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700182
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200183 if(PacketSize>1)
184 {
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200185 typedef Matrix<Scalar,3,3,ColMajor> Matrix33c;
Gael Guennebaude68e1652016-05-24 21:54:03 +0200186 typedef Matrix<Scalar,3,1,ColMajor> Vector3;
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200187 VERIFY(test_assign(Matrix33c().row(2),Matrix33c().row(1)+Matrix33c().row(1),
188 LinearTraversal,CompleteUnrolling));
Gael Guennebaude68e1652016-05-24 21:54:03 +0200189 VERIFY(test_assign(Vector3(),Vector3()+Vector3(),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100190 sizeof(Scalar)==16 ? InnerVectorizedTraversal : (EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal), CompleteUnrolling));
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200191 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100192 EIGEN_UNALIGNED_VECTORIZE ? (sizeof(Scalar)==16 ? InnerVectorizedTraversal : LinearVectorizedTraversal)
193 : (sizeof(Scalar)==16 ? SliceVectorizedTraversal : LinearTraversal),
194 ((!EIGEN_UNALIGNED_VECTORIZE) && (sizeof(Scalar)==16)) ? NoUnrolling : CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700195
196 VERIFY(test_assign(Matrix3(),Matrix3().cwiseProduct(Matrix3()),
197 LinearVectorizedTraversal,CompleteUnrolling));
198
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200199 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100200 sizeof(Scalar)==16 ? InnerVectorizedTraversal :
Gael Guennebaude68e1652016-05-24 21:54:03 +0200201 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal :
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100202 LinearTraversal,
Gael Guennebaude68e1652016-05-24 21:54:03 +0200203 NoUnrolling));
204
205 VERIFY(test_assign(Matrix11(), Matrix11()+Matrix11(),InnerVectorizedTraversal,CompleteUnrolling));
206
Benoit Steiner25d05c42016-04-12 14:13:25 -0700207
Gael Guennebaud91716f02018-09-21 14:32:24 +0200208 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 +0200209 (EIGEN_UNALIGNED_VECTORIZE) ? InnerVectorizedTraversal : DefaultTraversal, CompleteUnrolling|InnerUnrolling));
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200210
211 VERIFY(test_assign(Vector1(),Matrix11()*Vector1(),
212 InnerVectorizedTraversal,CompleteUnrolling));
213
214 VERIFY(test_assign(Matrix11(),Matrix11().lazyProduct(Matrix11()),
215 InnerVectorizedTraversal,InnerUnrolling+CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200216 }
Benoit Steiner25d05c42016-04-12 14:13:25 -0700217
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200218 VERIFY(test_redux(Vector1(),
219 LinearVectorizedTraversal,CompleteUnrolling));
220
Gael Guennebaud9c3aed92017-12-14 14:24:33 +0100221 VERIFY(test_redux(Vector1().array()*Vector1().array(),
222 LinearVectorizedTraversal,CompleteUnrolling));
223
224 VERIFY(test_redux((Vector1().array()*Vector1().array()).col(0),
225 LinearVectorizedTraversal,CompleteUnrolling));
226
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200227 VERIFY(test_redux(Matrix<Scalar,PacketSize,3>(),
228 LinearVectorizedTraversal,CompleteUnrolling));
229
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200230 VERIFY(test_redux(Matrix3(),
231 LinearVectorizedTraversal,CompleteUnrolling));
232
233 VERIFY(test_redux(Matrix44(),
234 LinearVectorizedTraversal,NoUnrolling));
235
Gael Guennebaudb00e48a2018-09-21 13:45:56 +0200236 if(PacketSize>1) {
237 VERIFY(test_redux(Matrix44().template block<(Matrix1::Flags&RowMajorBit)?4:PacketSize,(Matrix1::Flags&RowMajorBit)?PacketSize:4>(1,2),
238 SliceVectorizedTraversal,CompleteUnrolling));
239
240 VERIFY(test_redux(Matrix44().template block<(Matrix1::Flags&RowMajorBit)?2:PacketSize,(Matrix1::Flags&RowMajorBit)?PacketSize:2>(1,2),
241 DefaultTraversal,CompleteUnrolling));
242 }
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200243
244 VERIFY(test_redux(Matrix44c().template block<2*PacketSize,1>(1,2),
245 LinearVectorizedTraversal,CompleteUnrolling));
246
247 VERIFY(test_redux(Matrix44r().template block<1,2*PacketSize>(2,1),
248 LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200249
Gael Guennebaud501bc602011-05-19 21:52:40 +0200250 VERIFY((test_assign<
Gael Guennebaud1f502432015-08-06 15:31:07 +0200251 Map<Matrix22, AlignedMax, OuterStride<3*PacketSize> >,
Gael Guennebaud501bc602011-05-19 21:52:40 +0200252 Matrix22
253 >(InnerVectorizedTraversal,CompleteUnrolling)));
254
255 VERIFY((test_assign<
Gael Guennebaud1f502432015-08-06 15:31:07 +0200256 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 +0200257 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
Gael Guennebaud5ca24572016-07-06 22:25:24 +0200258 >(DefaultTraversal,PacketSize>=8?InnerUnrolling:CompleteUnrolling)));
Gael Guennebaud501bc602011-05-19 21:52:40 +0200259
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200260 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 +0100261 InnerVectorizedTraversal, CompleteUnrolling)));
Gael Guennebaud501bc602011-05-19 21:52:40 +0200262 #endif
263
264 VERIFY(test_assign(MatrixXX(10,10),MatrixXX(20,20).block(10,10,2,3),
265 SliceVectorizedTraversal,NoUnrolling));
266
267 VERIFY(test_redux(VectorX(10),
268 LinearVectorizedTraversal,NoUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200269 }
270};
271
272template<typename Scalar> struct vectorization_logic<Scalar,false>
273{
274 static void run() {}
275};
276
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200277template<typename Scalar, bool Enable = !internal::is_same<typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half,
278 typename internal::packet_traits<Scalar>::type>::value >
279struct vectorization_logic_half
280{
281 typedef internal::packet_traits<Scalar> PacketTraits;
282 typedef typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half PacketType;
283 enum {
284 PacketSize = internal::unpacket_traits<PacketType>::size
285 };
286 static void run()
287 {
288
289 typedef Matrix<Scalar,PacketSize,1> Vector1;
290 typedef Matrix<Scalar,PacketSize,PacketSize> Matrix11;
291 typedef Matrix<Scalar,5*PacketSize,7,ColMajor> Matrix57;
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200292 typedef Matrix<Scalar,3*PacketSize,5,ColMajor> Matrix35;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200293 typedef Matrix<Scalar,5*PacketSize,7,DontAlign|ColMajor> Matrix57u;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200294
295 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200296 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
297 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1)
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200298 > Matrix1;
299
300 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200301 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
302 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1),
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200303 DontAlign|((Matrix1::Flags&RowMajorBit)?RowMajor:ColMajor)> Matrix1u;
304
305 // this type is made such that it can only be vectorized when viewed as a linear 1D vector
306 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200307 (PacketSize==16 ? 4 : PacketSize==8 ? 4 : PacketSize==4 ? 6 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?2:3) : /*PacketSize==1 ?*/ 1),
308 (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 +0200309 > Matrix3;
310
311 #if !EIGEN_GCC_AND_ARCH_DOESNT_WANT_STACK_ALIGNMENT
312 VERIFY(test_assign(Vector1(),Vector1(),
313 InnerVectorizedTraversal,CompleteUnrolling));
314 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
315 InnerVectorizedTraversal,CompleteUnrolling));
Gael Guennebaude68e1652016-05-24 21:54:03 +0200316 VERIFY(test_assign(Vector1(),Vector1().template segment<PacketSize>(0).derived(),
317 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
318 VERIFY(test_assign(Vector1(),Scalar(2.1)*Vector1()-Vector1(),
319 InnerVectorizedTraversal,CompleteUnrolling));
320 VERIFY(test_assign(Vector1(),(Scalar(2.1)*Vector1().template segment<PacketSize>(0)-Vector1().template segment<PacketSize>(0)).derived(),
321 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200322 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
323 InnerVectorizedTraversal,CompleteUnrolling));
324 VERIFY(test_assign(Vector1(),Vector1().template cast<Scalar>(),
325 InnerVectorizedTraversal,CompleteUnrolling));
326
327
328 VERIFY(test_assign(Vector1(),Vector1(),
329 InnerVectorizedTraversal,CompleteUnrolling));
330 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
331 InnerVectorizedTraversal,CompleteUnrolling));
332 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
333 InnerVectorizedTraversal,CompleteUnrolling));
334
335 VERIFY(test_assign(Matrix57(),Matrix57()+Matrix57(),
336 InnerVectorizedTraversal,InnerUnrolling));
337
338 VERIFY(test_assign(Matrix57u(),Matrix57()+Matrix57(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200339 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearTraversal,
340 EIGEN_UNALIGNED_VECTORIZE ? InnerUnrolling : NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200341
342 VERIFY(test_assign(Matrix1u(),Matrix1()+Matrix1(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200343 EIGEN_UNALIGNED_VECTORIZE ? ((Matrix1::InnerSizeAtCompileTime % PacketSize)==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200344
345 if(PacketSize>1)
346 {
347 typedef Matrix<Scalar,3,3,ColMajor> Matrix33c;
348 VERIFY(test_assign(Matrix33c().row(2),Matrix33c().row(1)+Matrix33c().row(1),
349 LinearTraversal,CompleteUnrolling));
350 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100351 EIGEN_UNALIGNED_VECTORIZE ? (sizeof(Scalar)==16 ? InnerVectorizedTraversal : LinearVectorizedTraversal)
352 : (sizeof(Scalar)==16 ? SliceVectorizedTraversal : LinearTraversal),
353 ((!EIGEN_UNALIGNED_VECTORIZE) && (sizeof(Scalar)==16)) ? NoUnrolling : CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200354
355 VERIFY(test_assign(Matrix3(),Matrix3().cwiseQuotient(Matrix3()),
356 PacketTraits::HasDiv ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
357
358 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100359 sizeof(Scalar)==16 ? InnerVectorizedTraversal : (EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200360 NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200361
362 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 +0200363 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : DefaultTraversal,InnerUnrolling+CompleteUnrolling));
364
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200365
366 VERIFY(test_assign(Vector1(),Matrix11()*Vector1(),
367 InnerVectorizedTraversal,CompleteUnrolling));
368
369 VERIFY(test_assign(Matrix11(),Matrix11().lazyProduct(Matrix11()),
370 InnerVectorizedTraversal,InnerUnrolling+CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200371 }
372
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200373 VERIFY(test_redux(Vector1(),
374 LinearVectorizedTraversal,CompleteUnrolling));
375
376 VERIFY(test_redux(Matrix<Scalar,PacketSize,3>(),
377 LinearVectorizedTraversal,CompleteUnrolling));
378
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200379 VERIFY(test_redux(Matrix3(),
380 LinearVectorizedTraversal,CompleteUnrolling));
381
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200382 VERIFY(test_redux(Matrix35(),
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200383 LinearVectorizedTraversal,CompleteUnrolling));
384
Gael Guennebaudb00e48a2018-09-21 13:45:56 +0200385 VERIFY(test_redux(Matrix57().template block<PacketSize==1?2:PacketSize,3>(1,0),
386 SliceVectorizedTraversal,CompleteUnrolling));
387
388 if(PacketSize>1) {
389 VERIFY(test_redux(Matrix57().template block<PacketSize,2>(1,0),
390 DefaultTraversal,CompleteUnrolling));
391 }
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200392
393 VERIFY((test_assign<
394 Map<Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>, AlignedMax, InnerStride<3*PacketSize> >,
395 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
Gael Guennebaud91716f02018-09-21 14:32:24 +0200396 >(DefaultTraversal,PacketSize>4?InnerUnrolling:CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200397
398 VERIFY((test_assign(Matrix57(), Matrix<Scalar,5*PacketSize,3>()*Matrix<Scalar,3,7>(),
Gael Guennebaud91716f02018-09-21 14:32:24 +0200399 InnerVectorizedTraversal, InnerUnrolling+CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200400 #endif
401 }
402};
403
404template<typename Scalar> struct vectorization_logic_half<Scalar,false>
405{
406 static void run() {}
407};
408
Gael Guennebaud82f0ce22018-07-17 14:46:15 +0200409EIGEN_DECLARE_TEST(vectorization_logic)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000410{
411
412#ifdef EIGEN_VECTORIZE
413
Christoph Hertzberg4f440b82014-07-14 14:36:20 +0200414 CALL_SUBTEST( vectorization_logic<int>::run() );
Gael Guennebaud501bc602011-05-19 21:52:40 +0200415 CALL_SUBTEST( vectorization_logic<float>::run() );
416 CALL_SUBTEST( vectorization_logic<double>::run() );
417 CALL_SUBTEST( vectorization_logic<std::complex<float> >::run() );
418 CALL_SUBTEST( vectorization_logic<std::complex<double> >::run() );
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200419
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200420 CALL_SUBTEST( vectorization_logic_half<int>::run() );
421 CALL_SUBTEST( vectorization_logic_half<float>::run() );
422 CALL_SUBTEST( vectorization_logic_half<double>::run() );
423 CALL_SUBTEST( vectorization_logic_half<std::complex<float> >::run() );
424 CALL_SUBTEST( vectorization_logic_half<std::complex<double> >::run() );
425
Benoit Jacob47160402010-10-25 10:15:22 -0400426 if(internal::packet_traits<float>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200427 {
428 VERIFY(test_assign(Matrix<float,3,3>(),Matrix<float,3,3>()+Matrix<float,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200429 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200430
431 VERIFY(test_redux(Matrix<float,5,2>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200432 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200433 }
434
Benoit Jacob47160402010-10-25 10:15:22 -0400435 if(internal::packet_traits<double>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200436 {
437 VERIFY(test_assign(Matrix<double,3,3>(),Matrix<double,3,3>()+Matrix<double,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200438 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200439
440 VERIFY(test_redux(Matrix<double,7,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200441 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200442 }
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000443#endif // EIGEN_VECTORIZE
444
445}