blob: 8efee92a122a1f6fbb9d2d288f8baac799c0d5f8 [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{
Gael Guennebaud91716f02018-09-21 14:32:24 +020040 EIGEN_PREDICATE_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{
Gael Guennebaud91716f02018-09-21 14:32:24 +020065 EIGEN_PREDICATE_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(),
190 EIGEN_UNALIGNED_VECTORIZE ? (HalfPacketSize==1 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : (HalfPacketSize==1 ? InnerVectorizedTraversal : LinearTraversal), CompleteUnrolling));
Gael Guennebaudaa2b46a2010-07-23 16:29:29 +0200191 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200192 EIGEN_UNALIGNED_VECTORIZE ? (HalfPacketSize==1 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : (HalfPacketSize==1 ? SliceVectorizedTraversal : LinearTraversal),
193 ((!EIGEN_UNALIGNED_VECTORIZE) && HalfPacketSize==1) ? NoUnrolling : CompleteUnrolling));
Benoit Steiner25d05c42016-04-12 14:13:25 -0700194
195 VERIFY(test_assign(Matrix3(),Matrix3().cwiseProduct(Matrix3()),
196 LinearVectorizedTraversal,CompleteUnrolling));
197
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200198 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200199 HalfPacketSize==1 ? InnerVectorizedTraversal :
200 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal :
201 LinearTraversal,
202 NoUnrolling));
203
204 VERIFY(test_assign(Matrix11(), Matrix11()+Matrix11(),InnerVectorizedTraversal,CompleteUnrolling));
205
Benoit Steiner25d05c42016-04-12 14:13:25 -0700206
Gael Guennebaud91716f02018-09-21 14:32:24 +0200207 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 +0200208 (EIGEN_UNALIGNED_VECTORIZE) ? InnerVectorizedTraversal : DefaultTraversal, CompleteUnrolling|InnerUnrolling));
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200209
210 VERIFY(test_assign(Vector1(),Matrix11()*Vector1(),
211 InnerVectorizedTraversal,CompleteUnrolling));
212
213 VERIFY(test_assign(Matrix11(),Matrix11().lazyProduct(Matrix11()),
214 InnerVectorizedTraversal,InnerUnrolling+CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200215 }
Benoit Steiner25d05c42016-04-12 14:13:25 -0700216
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200217 VERIFY(test_redux(Vector1(),
218 LinearVectorizedTraversal,CompleteUnrolling));
219
Gael Guennebaud9c3aed92017-12-14 14:24:33 +0100220 VERIFY(test_redux(Vector1().array()*Vector1().array(),
221 LinearVectorizedTraversal,CompleteUnrolling));
222
223 VERIFY(test_redux((Vector1().array()*Vector1().array()).col(0),
224 LinearVectorizedTraversal,CompleteUnrolling));
225
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200226 VERIFY(test_redux(Matrix<Scalar,PacketSize,3>(),
227 LinearVectorizedTraversal,CompleteUnrolling));
228
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200229 VERIFY(test_redux(Matrix3(),
230 LinearVectorizedTraversal,CompleteUnrolling));
231
232 VERIFY(test_redux(Matrix44(),
233 LinearVectorizedTraversal,NoUnrolling));
234
Gael Guennebaudb00e48a2018-09-21 13:45:56 +0200235 if(PacketSize>1) {
236 VERIFY(test_redux(Matrix44().template block<(Matrix1::Flags&RowMajorBit)?4:PacketSize,(Matrix1::Flags&RowMajorBit)?PacketSize:4>(1,2),
237 SliceVectorizedTraversal,CompleteUnrolling));
238
239 VERIFY(test_redux(Matrix44().template block<(Matrix1::Flags&RowMajorBit)?2:PacketSize,(Matrix1::Flags&RowMajorBit)?PacketSize:2>(1,2),
240 DefaultTraversal,CompleteUnrolling));
241 }
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200242
243 VERIFY(test_redux(Matrix44c().template block<2*PacketSize,1>(1,2),
244 LinearVectorizedTraversal,CompleteUnrolling));
245
246 VERIFY(test_redux(Matrix44r().template block<1,2*PacketSize>(2,1),
247 LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200248
Gael Guennebaud501bc602011-05-19 21:52:40 +0200249 VERIFY((test_assign<
Gael Guennebaud1f502432015-08-06 15:31:07 +0200250 Map<Matrix22, AlignedMax, OuterStride<3*PacketSize> >,
Gael Guennebaud501bc602011-05-19 21:52:40 +0200251 Matrix22
252 >(InnerVectorizedTraversal,CompleteUnrolling)));
253
254 VERIFY((test_assign<
Gael Guennebaud1f502432015-08-06 15:31:07 +0200255 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 +0200256 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
Gael Guennebaud5ca24572016-07-06 22:25:24 +0200257 >(DefaultTraversal,PacketSize>=8?InnerUnrolling:CompleteUnrolling)));
Gael Guennebaud501bc602011-05-19 21:52:40 +0200258
Gael Guennebaud2606abe2014-04-18 21:14:40 +0200259 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 +0100260 InnerVectorizedTraversal, CompleteUnrolling)));
Gael Guennebaud501bc602011-05-19 21:52:40 +0200261 #endif
262
263 VERIFY(test_assign(MatrixXX(10,10),MatrixXX(20,20).block(10,10,2,3),
264 SliceVectorizedTraversal,NoUnrolling));
265
266 VERIFY(test_redux(VectorX(10),
267 LinearVectorizedTraversal,NoUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200268 }
269};
270
271template<typename Scalar> struct vectorization_logic<Scalar,false>
272{
273 static void run() {}
274};
275
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200276template<typename Scalar, bool Enable = !internal::is_same<typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half,
277 typename internal::packet_traits<Scalar>::type>::value >
278struct vectorization_logic_half
279{
280 typedef internal::packet_traits<Scalar> PacketTraits;
281 typedef typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half PacketType;
282 enum {
283 PacketSize = internal::unpacket_traits<PacketType>::size
284 };
285 static void run()
286 {
287
288 typedef Matrix<Scalar,PacketSize,1> Vector1;
289 typedef Matrix<Scalar,PacketSize,PacketSize> Matrix11;
290 typedef Matrix<Scalar,5*PacketSize,7,ColMajor> Matrix57;
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200291 typedef Matrix<Scalar,3*PacketSize,5,ColMajor> Matrix35;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200292 typedef Matrix<Scalar,5*PacketSize,7,DontAlign|ColMajor> Matrix57u;
293// typedef Matrix<Scalar,(Matrix11::Flags&RowMajorBit)?16:4*PacketSize,(Matrix11::Flags&RowMajorBit)?4*PacketSize:16> Matrix44;
294// typedef Matrix<Scalar,(Matrix11::Flags&RowMajorBit)?16:4*PacketSize,(Matrix11::Flags&RowMajorBit)?4*PacketSize:16,DontAlign|EIGEN_DEFAULT_MATRIX_STORAGE_ORDER_OPTION> Matrix44u;
295// typedef Matrix<Scalar,4*PacketSize,4*PacketSize,ColMajor> Matrix44c;
296// typedef Matrix<Scalar,4*PacketSize,4*PacketSize,RowMajor> Matrix44r;
297
298 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200299 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
300 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1)
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200301 > Matrix1;
302
303 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200304 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
305 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1),
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200306 DontAlign|((Matrix1::Flags&RowMajorBit)?RowMajor:ColMajor)> Matrix1u;
307
308 // this type is made such that it can only be vectorized when viewed as a linear 1D vector
309 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200310 (PacketSize==16 ? 4 : PacketSize==8 ? 4 : PacketSize==4 ? 6 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?2:3) : /*PacketSize==1 ?*/ 1),
311 (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 +0200312 > Matrix3;
313
314 #if !EIGEN_GCC_AND_ARCH_DOESNT_WANT_STACK_ALIGNMENT
315 VERIFY(test_assign(Vector1(),Vector1(),
316 InnerVectorizedTraversal,CompleteUnrolling));
317 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
318 InnerVectorizedTraversal,CompleteUnrolling));
Gael Guennebaude68e1652016-05-24 21:54:03 +0200319 VERIFY(test_assign(Vector1(),Vector1().template segment<PacketSize>(0).derived(),
320 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
321 VERIFY(test_assign(Vector1(),Scalar(2.1)*Vector1()-Vector1(),
322 InnerVectorizedTraversal,CompleteUnrolling));
323 VERIFY(test_assign(Vector1(),(Scalar(2.1)*Vector1().template segment<PacketSize>(0)-Vector1().template segment<PacketSize>(0)).derived(),
324 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200325 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
326 InnerVectorizedTraversal,CompleteUnrolling));
327 VERIFY(test_assign(Vector1(),Vector1().template cast<Scalar>(),
328 InnerVectorizedTraversal,CompleteUnrolling));
329
330
331 VERIFY(test_assign(Vector1(),Vector1(),
332 InnerVectorizedTraversal,CompleteUnrolling));
333 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
334 InnerVectorizedTraversal,CompleteUnrolling));
335 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
336 InnerVectorizedTraversal,CompleteUnrolling));
337
338 VERIFY(test_assign(Matrix57(),Matrix57()+Matrix57(),
339 InnerVectorizedTraversal,InnerUnrolling));
340
341 VERIFY(test_assign(Matrix57u(),Matrix57()+Matrix57(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200342 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearTraversal,
343 EIGEN_UNALIGNED_VECTORIZE ? InnerUnrolling : NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200344
345 VERIFY(test_assign(Matrix1u(),Matrix1()+Matrix1(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200346 EIGEN_UNALIGNED_VECTORIZE ? ((Matrix1::InnerSizeAtCompileTime % PacketSize)==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200347
348 if(PacketSize>1)
349 {
350 typedef Matrix<Scalar,3,3,ColMajor> Matrix33c;
351 VERIFY(test_assign(Matrix33c().row(2),Matrix33c().row(1)+Matrix33c().row(1),
352 LinearTraversal,CompleteUnrolling));
353 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200354 EIGEN_UNALIGNED_VECTORIZE ? (PacketSize==1 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200355
356 VERIFY(test_assign(Matrix3(),Matrix3().cwiseQuotient(Matrix3()),
357 PacketTraits::HasDiv ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
358
359 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200360 EIGEN_UNALIGNED_VECTORIZE ? (PacketSize==1 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,
361 NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200362
363 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 +0200364 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : DefaultTraversal,InnerUnrolling+CompleteUnrolling));
365
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200366
367 VERIFY(test_assign(Vector1(),Matrix11()*Vector1(),
368 InnerVectorizedTraversal,CompleteUnrolling));
369
370 VERIFY(test_assign(Matrix11(),Matrix11().lazyProduct(Matrix11()),
371 InnerVectorizedTraversal,InnerUnrolling+CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200372 }
373
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200374 VERIFY(test_redux(Vector1(),
375 LinearVectorizedTraversal,CompleteUnrolling));
376
377 VERIFY(test_redux(Matrix<Scalar,PacketSize,3>(),
378 LinearVectorizedTraversal,CompleteUnrolling));
379
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200380 VERIFY(test_redux(Matrix3(),
381 LinearVectorizedTraversal,CompleteUnrolling));
382
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200383 VERIFY(test_redux(Matrix35(),
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200384 LinearVectorizedTraversal,CompleteUnrolling));
385
Gael Guennebaudb00e48a2018-09-21 13:45:56 +0200386 VERIFY(test_redux(Matrix57().template block<PacketSize==1?2:PacketSize,3>(1,0),
387 SliceVectorizedTraversal,CompleteUnrolling));
388
389 if(PacketSize>1) {
390 VERIFY(test_redux(Matrix57().template block<PacketSize,2>(1,0),
391 DefaultTraversal,CompleteUnrolling));
392 }
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200393
394 VERIFY((test_assign<
395 Map<Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>, AlignedMax, InnerStride<3*PacketSize> >,
396 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
Gael Guennebaud91716f02018-09-21 14:32:24 +0200397 >(DefaultTraversal,PacketSize>4?InnerUnrolling:CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200398
399 VERIFY((test_assign(Matrix57(), Matrix<Scalar,5*PacketSize,3>()*Matrix<Scalar,3,7>(),
Gael Guennebaud91716f02018-09-21 14:32:24 +0200400 InnerVectorizedTraversal, InnerUnrolling+CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200401 #endif
402 }
403};
404
405template<typename Scalar> struct vectorization_logic_half<Scalar,false>
406{
407 static void run() {}
408};
409
Gael Guennebaud82f0ce22018-07-17 14:46:15 +0200410EIGEN_DECLARE_TEST(vectorization_logic)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000411{
412
413#ifdef EIGEN_VECTORIZE
414
Christoph Hertzberg4f440b82014-07-14 14:36:20 +0200415 CALL_SUBTEST( vectorization_logic<int>::run() );
Gael Guennebaud501bc602011-05-19 21:52:40 +0200416 CALL_SUBTEST( vectorization_logic<float>::run() );
417 CALL_SUBTEST( vectorization_logic<double>::run() );
418 CALL_SUBTEST( vectorization_logic<std::complex<float> >::run() );
419 CALL_SUBTEST( vectorization_logic<std::complex<double> >::run() );
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200420
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200421 CALL_SUBTEST( vectorization_logic_half<int>::run() );
422 CALL_SUBTEST( vectorization_logic_half<float>::run() );
423 CALL_SUBTEST( vectorization_logic_half<double>::run() );
424 CALL_SUBTEST( vectorization_logic_half<std::complex<float> >::run() );
425 CALL_SUBTEST( vectorization_logic_half<std::complex<double> >::run() );
426
Benoit Jacob47160402010-10-25 10:15:22 -0400427 if(internal::packet_traits<float>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200428 {
429 VERIFY(test_assign(Matrix<float,3,3>(),Matrix<float,3,3>()+Matrix<float,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200430 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200431
432 VERIFY(test_redux(Matrix<float,5,2>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200433 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200434 }
435
Benoit Jacob47160402010-10-25 10:15:22 -0400436 if(internal::packet_traits<double>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200437 {
438 VERIFY(test_assign(Matrix<double,3,3>(),Matrix<double,3,3>()+Matrix<double,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200439 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200440
441 VERIFY(test_redux(Matrix<double,7,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200442 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200443 }
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000444#endif // EIGEN_VECTORIZE
445
446}