blob: 602e9f15c33e73b38b6e57da8b8865e9758af064 [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));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200261 }
262};
263
264template<typename Scalar> struct vectorization_logic<Scalar,false>
265{
266 static void run() {}
267};
268
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200269template<typename Scalar, bool Enable = !internal::is_same<typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half,
270 typename internal::packet_traits<Scalar>::type>::value >
271struct vectorization_logic_half
272{
273 typedef internal::packet_traits<Scalar> PacketTraits;
274 typedef typename internal::unpacket_traits<typename internal::packet_traits<Scalar>::type>::half PacketType;
275 enum {
276 PacketSize = internal::unpacket_traits<PacketType>::size
277 };
278 static void run()
279 {
280
281 typedef Matrix<Scalar,PacketSize,1> Vector1;
282 typedef Matrix<Scalar,PacketSize,PacketSize> Matrix11;
283 typedef Matrix<Scalar,5*PacketSize,7,ColMajor> Matrix57;
Gael Guennebaud512ba0a2016-04-13 18:16:35 +0200284 typedef Matrix<Scalar,3*PacketSize,5,ColMajor> Matrix35;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200285 typedef Matrix<Scalar,5*PacketSize,7,DontAlign|ColMajor> Matrix57u;
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200286
287 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200288 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
289 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1)
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200290 > Matrix1;
291
292 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200293 (PacketSize==16 ? 8 : PacketSize==8 ? 4 : PacketSize==4 ? 2 : PacketSize==2 ? 1 : /*PacketSize==1 ?*/ 1),
294 (PacketSize==16 ? 2 : PacketSize==8 ? 2 : PacketSize==4 ? 2 : PacketSize==2 ? 2 : /*PacketSize==1 ?*/ 1),
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200295 DontAlign|((Matrix1::Flags&RowMajorBit)?RowMajor:ColMajor)> Matrix1u;
296
297 // this type is made such that it can only be vectorized when viewed as a linear 1D vector
298 typedef Matrix<Scalar,
Gael Guennebaud91716f02018-09-21 14:32:24 +0200299 (PacketSize==16 ? 4 : PacketSize==8 ? 4 : PacketSize==4 ? 6 : PacketSize==2 ? ((Matrix11::Flags&RowMajorBit)?2:3) : /*PacketSize==1 ?*/ 1),
300 (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 +0200301 > Matrix3;
302
303 #if !EIGEN_GCC_AND_ARCH_DOESNT_WANT_STACK_ALIGNMENT
304 VERIFY(test_assign(Vector1(),Vector1(),
305 InnerVectorizedTraversal,CompleteUnrolling));
306 VERIFY(test_assign(Vector1(),Vector1()+Vector1(),
307 InnerVectorizedTraversal,CompleteUnrolling));
Gael Guennebaude68e1652016-05-24 21:54:03 +0200308 VERIFY(test_assign(Vector1(),Vector1().template segment<PacketSize>(0).derived(),
309 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
310 VERIFY(test_assign(Vector1(),Scalar(2.1)*Vector1()-Vector1(),
311 InnerVectorizedTraversal,CompleteUnrolling));
312 VERIFY(test_assign(Vector1(),(Scalar(2.1)*Vector1().template segment<PacketSize>(0)-Vector1().template segment<PacketSize>(0)).derived(),
313 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearVectorizedTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200314 VERIFY(test_assign(Vector1(),Vector1().cwiseProduct(Vector1()),
315 InnerVectorizedTraversal,CompleteUnrolling));
316 VERIFY(test_assign(Vector1(),Vector1().template cast<Scalar>(),
317 InnerVectorizedTraversal,CompleteUnrolling));
318
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200319 VERIFY(test_assign(Matrix57(),Matrix57()+Matrix57(),
320 InnerVectorizedTraversal,InnerUnrolling));
321
322 VERIFY(test_assign(Matrix57u(),Matrix57()+Matrix57(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200323 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : LinearTraversal,
324 EIGEN_UNALIGNED_VECTORIZE ? InnerUnrolling : NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200325
326 VERIFY(test_assign(Matrix1u(),Matrix1()+Matrix1(),
Rasmus Munk Larsen13fb5ab2021-06-15 09:09:31 -0700327 EIGEN_UNALIGNED_VECTORIZE ? ((int(Matrix1::InnerSizeAtCompileTime) % int(PacketSize))==0 ? InnerVectorizedTraversal : LinearVectorizedTraversal) : LinearTraversal,CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200328
329 if(PacketSize>1)
330 {
331 typedef Matrix<Scalar,3,3,ColMajor> Matrix33c;
332 VERIFY(test_assign(Matrix33c().row(2),Matrix33c().row(1)+Matrix33c().row(1),
333 LinearTraversal,CompleteUnrolling));
334 VERIFY(test_assign(Matrix33c().col(0),Matrix33c().col(1)+Matrix33c().col(1),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100335 EIGEN_UNALIGNED_VECTORIZE ? (sizeof(Scalar)==16 ? InnerVectorizedTraversal : LinearVectorizedTraversal)
336 : (sizeof(Scalar)==16 ? SliceVectorizedTraversal : LinearTraversal),
337 ((!EIGEN_UNALIGNED_VECTORIZE) && (sizeof(Scalar)==16)) ? NoUnrolling : CompleteUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200338
339 VERIFY(test_assign(Matrix3(),Matrix3().cwiseQuotient(Matrix3()),
Jakub Lichmandc5b1f72021-08-25 19:38:23 +0000340 PacketTraits::HasDiv ? LinearVectorizedTraversal : LinearTraversal,
341 PacketTraits::HasDiv ? CompleteUnrolling : NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200342
343 VERIFY(test_assign(Matrix<Scalar,17,17>(),Matrix<Scalar,17,17>()+Matrix<Scalar,17,17>(),
Gael Guennebaudaab749b2018-12-06 16:55:00 +0100344 sizeof(Scalar)==16 ? InnerVectorizedTraversal : (EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200345 NoUnrolling));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200346
347 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 +0200348 EIGEN_UNALIGNED_VECTORIZE ? InnerVectorizedTraversal : DefaultTraversal,InnerUnrolling+CompleteUnrolling));
349
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
Gael Guennebaudb00e48a2018-09-21 13:45:56 +0200370 VERIFY(test_redux(Matrix57().template block<PacketSize==1?2:PacketSize,3>(1,0),
371 SliceVectorizedTraversal,CompleteUnrolling));
372
373 if(PacketSize>1) {
374 VERIFY(test_redux(Matrix57().template block<PacketSize,2>(1,0),
375 DefaultTraversal,CompleteUnrolling));
376 }
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200377
378 VERIFY((test_assign<
379 Map<Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>, AlignedMax, InnerStride<3*PacketSize> >,
380 Matrix<Scalar,EIGEN_PLAIN_ENUM_MAX(2,PacketSize),EIGEN_PLAIN_ENUM_MAX(2,PacketSize)>
Gael Guennebaud91716f02018-09-21 14:32:24 +0200381 >(DefaultTraversal,PacketSize>4?InnerUnrolling:CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200382
383 VERIFY((test_assign(Matrix57(), Matrix<Scalar,5*PacketSize,3>()*Matrix<Scalar,3,7>(),
Gael Guennebaud91716f02018-09-21 14:32:24 +0200384 InnerVectorizedTraversal, InnerUnrolling+CompleteUnrolling)));
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200385 #endif
386 }
387};
388
389template<typename Scalar> struct vectorization_logic_half<Scalar,false>
390{
391 static void run() {}
392};
393
Gael Guennebaud82f0ce22018-07-17 14:46:15 +0200394EIGEN_DECLARE_TEST(vectorization_logic)
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000395{
396
397#ifdef EIGEN_VECTORIZE
398
Christoph Hertzberg4f440b82014-07-14 14:36:20 +0200399 CALL_SUBTEST( vectorization_logic<int>::run() );
Gael Guennebaud501bc602011-05-19 21:52:40 +0200400 CALL_SUBTEST( vectorization_logic<float>::run() );
401 CALL_SUBTEST( vectorization_logic<double>::run() );
402 CALL_SUBTEST( vectorization_logic<std::complex<float> >::run() );
403 CALL_SUBTEST( vectorization_logic<std::complex<double> >::run() );
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200404
Gael Guennebaudfebcce32015-08-07 20:05:31 +0200405 CALL_SUBTEST( vectorization_logic_half<int>::run() );
406 CALL_SUBTEST( vectorization_logic_half<float>::run() );
407 CALL_SUBTEST( vectorization_logic_half<double>::run() );
408 CALL_SUBTEST( vectorization_logic_half<std::complex<float> >::run() );
409 CALL_SUBTEST( vectorization_logic_half<std::complex<double> >::run() );
410
Benoit Jacob47160402010-10-25 10:15:22 -0400411 if(internal::packet_traits<float>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200412 {
413 VERIFY(test_assign(Matrix<float,3,3>(),Matrix<float,3,3>()+Matrix<float,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200414 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200415
416 VERIFY(test_redux(Matrix<float,5,2>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200417 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200418 }
419
Benoit Jacob47160402010-10-25 10:15:22 -0400420 if(internal::packet_traits<double>::Vectorizable)
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200421 {
422 VERIFY(test_assign(Matrix<double,3,3>(),Matrix<double,3,3>()+Matrix<double,3,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200423 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : LinearTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200424
425 VERIFY(test_redux(Matrix<double,7,3>(),
Gael Guennebaude68e1652016-05-24 21:54:03 +0200426 EIGEN_UNALIGNED_VECTORIZE ? LinearVectorizedTraversal : DefaultTraversal,CompleteUnrolling));
Gael Guennebaud51ec1882010-07-08 23:30:16 +0200427 }
Gael Guennebaudcb71dc42009-01-07 22:20:03 +0000428#endif // EIGEN_VECTORIZE
429
430}