WPILibC++ 2023.4.3
BlasUtil.h
Go to the documentation of this file.
1// This file is part of Eigen, a lightweight C++ template library
2// for linear algebra.
3//
4// Copyright (C) 2009-2010 Gael Guennebaud <gael.guennebaud@inria.fr>
5//
6// 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/.
9
10#ifndef EIGEN_BLASUTIL_H
11#define EIGEN_BLASUTIL_H
12
13// This file contains many lightweight helper classes used to
14// implement and control fast level 2 and level 3 BLAS-like routines.
15
16namespace Eigen {
17
18namespace internal {
19
20// forward declarations
21template<typename LhsScalar, typename RhsScalar, typename Index, typename DataMapper, int mr, int nr, bool ConjugateLhs=false, bool ConjugateRhs=false>
22struct gebp_kernel;
23
24template<typename Scalar, typename Index, typename DataMapper, int nr, int StorageOrder, bool Conjugate = false, bool PanelMode=false>
26
27template<typename Scalar, typename Index, typename DataMapper, int Pack1, int Pack2, typename Packet, int StorageOrder, bool Conjugate = false, bool PanelMode = false>
29
30template<
31 typename Index,
32 typename LhsScalar, int LhsStorageOrder, bool ConjugateLhs,
33 typename RhsScalar, int RhsStorageOrder, bool ConjugateRhs,
34 int ResStorageOrder, int ResInnerStride>
36
37template<typename Index,
38 typename LhsScalar, typename LhsMapper, int LhsStorageOrder, bool ConjugateLhs,
39 typename RhsScalar, typename RhsMapper, bool ConjugateRhs, int Version=Specialized>
41
42template<typename From,typename To> struct get_factor {
43 EIGEN_DEVICE_FUNC static EIGEN_STRONG_INLINE To run(const From& x) { return To(x); }
44};
45
46template<typename Scalar> struct get_factor<Scalar,typename NumTraits<Scalar>::Real> {
48 static EIGEN_STRONG_INLINE typename NumTraits<Scalar>::Real run(const Scalar& x) { return numext::real(x); }
49};
50
51
52template<typename Scalar, typename Index>
54 public:
56
58 return m_data[i];
59 }
60 template <typename Packet, int AlignmentType>
62 return ploadt<Packet, AlignmentType>(m_data + i);
63 }
64
65 template <typename Packet>
67 return (UIntPtr(m_data+i)%sizeof(Packet))==0;
68 }
69
70 protected:
71 Scalar* m_data;
72};
73
74template<typename Scalar, typename Index, int AlignmentType, int Incr=1>
76
77template<typename Scalar, typename Index, int AlignmentType>
79{
80public:
82 : m_data(data)
83 {
85 eigen_assert(incr==1);
86 }
87
89 internal::prefetch(&operator()(i));
90 }
91
93 return m_data[i];
94 }
95
96 template<typename PacketType>
98 return ploadt<PacketType, AlignmentType>(m_data + i);
99 }
100
101 template<typename PacketType>
102 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void storePacket(Index i, const PacketType &p) const {
103 pstoret<Scalar, PacketType, AlignmentType>(m_data + i, p);
104 }
105
106protected:
107 Scalar *m_data;
108};
109
110// Lightweight helper class to access matrix coefficients.
111template<typename Scalar, typename Index, int StorageOrder, int AlignmentType = Unaligned, int Incr = 1>
112class blas_data_mapper;
113
114// TMP to help PacketBlock store implementation.
115// There's currently no known use case for PacketBlock load.
116// The default implementation assumes ColMajor order.
117// It always store each packet sequentially one `stride` apart.
118template<typename Index, typename Scalar, typename Packet, int n, int idx, int StorageOrder>
120{
121 PacketBlockManagement<Index, Scalar, Packet, n, idx - 1, StorageOrder> pbm;
122 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(Scalar *to, const Index stride, Index i, Index j, const PacketBlock<Packet, n> &block) const {
123 pbm.store(to, stride, i, j, block);
124 pstoreu<Scalar>(to + i + (j + idx)*stride, block.packet[idx]);
125 }
126};
127
128// PacketBlockManagement specialization to take care of RowMajor order without ifs.
129template<typename Index, typename Scalar, typename Packet, int n, int idx>
130struct PacketBlockManagement<Index, Scalar, Packet, n, idx, RowMajor>
131{
132 PacketBlockManagement<Index, Scalar, Packet, n, idx - 1, RowMajor> pbm;
133 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(Scalar *to, const Index stride, Index i, Index j, const PacketBlock<Packet, n> &block) const {
134 pbm.store(to, stride, i, j, block);
135 pstoreu<Scalar>(to + j + (i + idx)*stride, block.packet[idx]);
136 }
137};
138
139template<typename Index, typename Scalar, typename Packet, int n, int StorageOrder>
140struct PacketBlockManagement<Index, Scalar, Packet, n, -1, StorageOrder>
141{
142 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(Scalar *to, const Index stride, Index i, Index j, const PacketBlock<Packet, n> &block) const {
144 EIGEN_UNUSED_VARIABLE(stride);
148 }
149};
150
151template<typename Index, typename Scalar, typename Packet, int n>
152struct PacketBlockManagement<Index, Scalar, Packet, n, -1, RowMajor>
153{
154 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(Scalar *to, const Index stride, Index i, Index j, const PacketBlock<Packet, n> &block) const {
156 EIGEN_UNUSED_VARIABLE(stride);
160 }
161};
162
163template<typename Scalar, typename Index, int StorageOrder, int AlignmentType>
164class blas_data_mapper<Scalar,Index,StorageOrder,AlignmentType,1>
165{
166public:
169
171 : m_data(data), m_stride(stride)
172 {
174 eigen_assert(incr==1);
175 }
176
180 }
181
183 return LinearMapper(&operator()(i, j));
184 }
185
187 return VectorMapper(&operator()(i, j));
188 }
189
190
193 return m_data[StorageOrder==RowMajor ? j + i*m_stride : i + j*m_stride];
194 }
195
196 template<typename PacketType>
198 return ploadt<PacketType, AlignmentType>(&operator()(i, j));
199 }
200
201 template <typename PacketT, int AlignmentT>
203 return ploadt<PacketT, AlignmentT>(&operator()(i, j));
204 }
205
206 template<typename SubPacket>
207 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void scatterPacket(Index i, Index j, const SubPacket &p) const {
208 pscatter<Scalar, SubPacket>(&operator()(i, j), p, m_stride);
209 }
210
211 template<typename SubPacket>
213 return pgather<Scalar, SubPacket>(&operator()(i, j), m_stride);
214 }
215
216 EIGEN_DEVICE_FUNC const Index stride() const { return m_stride; }
217 EIGEN_DEVICE_FUNC const Scalar* data() const { return m_data; }
218
220 if (UIntPtr(m_data)%sizeof(Scalar)) {
221 return -1;
222 }
224 }
225
226 template<typename SubPacket, int n>
228 PacketBlockManagement<Index, Scalar, SubPacket, n, n-1, StorageOrder> pbm;
229 pbm.store(m_data, m_stride, i, j, block);
230 }
231protected:
234};
235
236// Implementation of non-natural increment (i.e. inner-stride != 1)
237// The exposed API is not complete yet compared to the Incr==1 case
238// because some features makes less sense in this case.
239template<typename Scalar, typename Index, int AlignmentType, int Incr>
241{
242public:
244
246 internal::prefetch(&operator()(i));
247 }
248
250 return m_data[i*m_incr.value()];
251 }
252
253 template<typename PacketType>
255 return pgather<Scalar,PacketType>(m_data + i*m_incr.value(), m_incr.value());
256 }
257
258 template<typename PacketType>
259 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void storePacket(Index i, const PacketType &p) const {
260 pscatter<Scalar, PacketType>(m_data + i*m_incr.value(), p, m_incr.value());
261 }
262
263protected:
264 Scalar *m_data;
266};
267
268template<typename Scalar, typename Index, int StorageOrder, int AlignmentType,int Incr>
270{
271public:
273
275
278 return blas_data_mapper(&operator()(i, j), m_stride, m_incr.value());
279 }
280
282 return LinearMapper(&operator()(i, j), m_incr.value());
283 }
284
287 return m_data[StorageOrder==RowMajor ? j*m_incr.value() + i*m_stride : i*m_incr.value() + j*m_stride];
288 }
289
290 template<typename PacketType>
292 return pgather<Scalar,PacketType>(&operator()(i, j),m_incr.value());
293 }
294
295 template <typename PacketT, int AlignmentT>
297 return pgather<Scalar,PacketT>(&operator()(i, j),m_incr.value());
298 }
299
300 template<typename SubPacket>
301 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void scatterPacket(Index i, Index j, const SubPacket &p) const {
302 pscatter<Scalar, SubPacket>(&operator()(i, j), p, m_stride);
303 }
304
305 template<typename SubPacket>
307 return pgather<Scalar, SubPacket>(&operator()(i, j), m_stride);
308 }
309
310 // storePacketBlock_helper defines a way to access values inside the PacketBlock, this is essentially required by the Complex types.
311 template<typename SubPacket, typename ScalarT, int n, int idx>
313 {
314 storePacketBlock_helper<SubPacket, ScalarT, n, idx-1> spbh;
316 spbh.store(sup, i,j,block);
317 for(int l = 0; l < unpacket_traits<SubPacket>::size; l++)
318 {
319 ScalarT *v = &sup->operator()(i+l, j+idx);
320 *v = block.packet[idx][l];
321 }
322 }
323 };
324
325 template<typename SubPacket, int n, int idx>
326 struct storePacketBlock_helper<SubPacket, std::complex<float>, n, idx>
327 {
330 spbh.store(sup,i,j,block);
331 for(int l = 0; l < unpacket_traits<SubPacket>::size; l++)
332 {
333 std::complex<float> *v = &sup->operator()(i+l, j+idx);
334 v->real(block.packet[idx].v[2*l+0]);
335 v->imag(block.packet[idx].v[2*l+1]);
336 }
337 }
338 };
339
340 template<typename SubPacket, int n, int idx>
341 struct storePacketBlock_helper<SubPacket, std::complex<double>, n, idx>
342 {
345 spbh.store(sup,i,j,block);
346 for(int l = 0; l < unpacket_traits<SubPacket>::size; l++)
347 {
348 std::complex<double> *v = &sup->operator()(i+l, j+idx);
349 v->real(block.packet[idx].v[2*l+0]);
350 v->imag(block.packet[idx].v[2*l+1]);
351 }
352 }
353 };
354
355 template<typename SubPacket, typename ScalarT, int n>
356 struct storePacketBlock_helper<SubPacket, ScalarT, n, -1>
357 {
359 }
360 };
361
362 template<typename SubPacket, int n>
363 struct storePacketBlock_helper<SubPacket, std::complex<float>, n, -1>
364 {
366 }
367 };
368
369 template<typename SubPacket, int n>
370 struct storePacketBlock_helper<SubPacket, std::complex<double>, n, -1>
371 {
373 }
374 };
375 // This function stores a PacketBlock on m_data, this approach is really quite slow compare to Incr=1 and should be avoided when possible.
376 template<typename SubPacket, int n>
378 storePacketBlock_helper<SubPacket, Scalar, n, n-1> spb;
379 spb.store(this, i,j,block);
380 }
381protected:
385};
386
387// lightweight helper class to access matrix coefficients (const version)
388template<typename Scalar, typename Index, int StorageOrder>
389class const_blas_data_mapper : public blas_data_mapper<const Scalar, Index, StorageOrder> {
390 public:
391 EIGEN_ALWAYS_INLINE const_blas_data_mapper(const Scalar *data, Index stride) : blas_data_mapper<const Scalar, Index, StorageOrder>(data, stride) {}
392
394 return const_blas_data_mapper<Scalar, Index, StorageOrder>(&(this->operator()(i, j)), this->m_stride);
395 }
396};
397
398
399/* Helper class to analyze the factors of a Product expression.
400 * In particular it allows to pop out operator-, scalar multiples,
401 * and conjugate */
402template<typename XprType> struct blas_traits
403{
405 typedef const XprType& ExtractType;
406 typedef XprType _ExtractType;
407 enum {
411 HasUsableDirectAccess = ( (int(XprType::Flags)&DirectAccessBit)
412 && ( bool(XprType::IsVectorAtCompileTime)
414 ) ? 1 : 0,
415 HasScalarFactor = false
416 };
417 typedef typename conditional<bool(HasUsableDirectAccess),
419 typename _ExtractType::PlainObject
421 static inline EIGEN_DEVICE_FUNC ExtractType extract(const XprType& x) { return x; }
422 static inline EIGEN_DEVICE_FUNC const Scalar extractScalarFactor(const XprType&) { return Scalar(1); }
423};
424
425// pop conjugate
426template<typename Scalar, typename NestedXpr>
428 : blas_traits<NestedXpr>
429{
433
434 enum {
436 NeedToConjugate = Base::NeedToConjugate ? 0 : IsComplex
437 };
438 static inline ExtractType extract(const XprType& x) { return Base::extract(x.nestedExpression()); }
439 static inline Scalar extractScalarFactor(const XprType& x) { return conj(Base::extractScalarFactor(x.nestedExpression())); }
440};
441
442// pop scalar multiple
443template<typename Scalar, typename NestedXpr, typename Plain>
445 : blas_traits<NestedXpr>
446{
447 enum {
448 HasScalarFactor = true
449 };
453 static inline EIGEN_DEVICE_FUNC ExtractType extract(const XprType& x) { return Base::extract(x.rhs()); }
455 { return x.lhs().functor().m_other * Base::extractScalarFactor(x.rhs()); }
456};
457template<typename Scalar, typename NestedXpr, typename Plain>
459 : blas_traits<NestedXpr>
460{
461 enum {
462 HasScalarFactor = true
463 };
467 static inline ExtractType extract(const XprType& x) { return Base::extract(x.lhs()); }
468 static inline Scalar extractScalarFactor(const XprType& x)
469 { return Base::extractScalarFactor(x.lhs()) * x.rhs().functor().m_other; }
470};
471template<typename Scalar, typename Plain1, typename Plain2>
473 const CwiseNullaryOp<scalar_constant_op<Scalar>,Plain2> > >
474 : blas_traits<CwiseNullaryOp<scalar_constant_op<Scalar>,Plain1> >
475{};
476
477// pop opposite
478template<typename Scalar, typename NestedXpr>
480 : blas_traits<NestedXpr>
481{
482 enum {
483 HasScalarFactor = true
484 };
488 static inline ExtractType extract(const XprType& x) { return Base::extract(x.nestedExpression()); }
489 static inline Scalar extractScalarFactor(const XprType& x)
490 { return - Base::extractScalarFactor(x.nestedExpression()); }
491};
492
493// pop/push transpose
494template<typename NestedXpr>
495struct blas_traits<Transpose<NestedXpr> >
496 : blas_traits<NestedXpr>
497{
498 typedef typename NestedXpr::Scalar Scalar;
501 typedef Transpose<const typename Base::_ExtractType> ExtractType; // const to get rid of a compile error; anyway blas traits are only used on the RHS
503 typedef typename conditional<bool(Base::HasUsableDirectAccess),
505 typename ExtractType::PlainObject
507 enum {
508 IsTransposed = Base::IsTransposed ? 0 : 1
509 };
510 static inline ExtractType extract(const XprType& x) { return ExtractType(Base::extract(x.nestedExpression())); }
511 static inline Scalar extractScalarFactor(const XprType& x) { return Base::extractScalarFactor(x.nestedExpression()); }
512};
513
514template<typename T>
515struct blas_traits<const T>
516 : blas_traits<T>
517{};
518
519template<typename T, bool HasUsableDirectAccess=blas_traits<T>::HasUsableDirectAccess>
521 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE static const typename T::Scalar* run(const T& m)
522 {
523 return blas_traits<T>::extract(m).data();
524 }
525};
526
527template<typename T>
528struct extract_data_selector<T,false> {
529 static typename T::Scalar* run(const T&) { return 0; }
530};
531
532template<typename T>
533EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE const typename T::Scalar* extract_data(const T& m)
534{
536}
537
538/**
539 * \c combine_scalar_factors extracts and multiplies factors from GEMM and GEMV products.
540 * There is a specialization for booleans
541 */
542template<typename ResScalar, typename Lhs, typename Rhs>
544{
545 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE static ResScalar run(const Lhs& lhs, const Rhs& rhs)
546 {
548 }
549 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE static ResScalar run(const ResScalar& alpha, const Lhs& lhs, const Rhs& rhs)
550 {
552 }
553};
554template<typename Lhs, typename Rhs>
555struct combine_scalar_factors_impl<bool, Lhs, Rhs>
556{
557 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE static bool run(const Lhs& lhs, const Rhs& rhs)
558 {
560 }
561 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE static bool run(const bool& alpha, const Lhs& lhs, const Rhs& rhs)
562 {
564 }
565};
566
567template<typename ResScalar, typename Lhs, typename Rhs>
568EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE ResScalar combine_scalar_factors(const ResScalar& alpha, const Lhs& lhs, const Rhs& rhs)
569{
571}
572template<typename ResScalar, typename Lhs, typename Rhs>
573EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE ResScalar combine_scalar_factors(const Lhs& lhs, const Rhs& rhs)
574{
576}
577
578
579} // end namespace internal
580
581} // end namespace Eigen
582
583#endif // EIGEN_BLASUTIL_H
EIGEN_DOC_BLOCK_ADDONS_NOT_INNER_PANEL EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE FixedBlockXpr< internal::get_fixed_value< NRowsType >::value, internal::get_fixed_value< NColsType >::value >::Type block(Index startRow, Index startCol, NRowsType blockRows, NColsType blockCols)
Definition: BlockMethods.h:96
EIGEN_DEVICE_FUNC RealReturnType real() const
Definition: CommonCwiseUnaryOps.h:100
#define EIGEN_RESTRICT
Definition: Macros.h:1170
#define EIGEN_ALWAYS_INLINE
Definition: Macros.h:942
#define EIGEN_UNUSED_VARIABLE(var)
Definition: Macros.h:1086
#define EIGEN_DEVICE_FUNC
Definition: Macros.h:986
#define EIGEN_ONLY_USED_FOR_DEBUG(x)
Definition: Macros.h:1059
#define eigen_assert(x)
Definition: Macros.h:1047
#define EIGEN_STRONG_INLINE
Definition: Macros.h:927
Generic expression where a coefficient-wise binary operator is applied to two expressions.
Definition: CwiseBinaryOp.h:84
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const _RhsNested & rhs() const
Definition: CwiseBinaryOp.h:135
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const _LhsNested & lhs() const
Definition: CwiseBinaryOp.h:132
Generic expression of a matrix where all coefficients are defined by a functor.
Definition: CwiseNullaryOp.h:61
Generic expression where a coefficient-wise unary operator is applied to an expression.
Definition: CwiseUnaryOp.h:56
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const internal::remove_all< XprTypeNested >::type & nestedExpression() const
Definition: CwiseUnaryOp.h:80
Expression of the transpose of a matrix.
Definition: Transpose.h:54
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const internal::remove_all< MatrixTypeNested >::type & nestedExpression() const
Definition: Transpose.h:76
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void storePacket(Index i, const PacketType &p) const
Definition: BlasUtil.h:102
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE BlasLinearMapper(Scalar *data, Index incr=1)
Definition: BlasUtil.h:81
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Scalar & operator()(Index i) const
Definition: BlasUtil.h:92
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketType loadPacket(Index i) const
Definition: BlasUtil.h:97
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void prefetch(int i) const
Definition: BlasUtil.h:88
Definition: BlasUtil.h:241
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void prefetch(int i) const
Definition: BlasUtil.h:245
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketType loadPacket(Index i) const
Definition: BlasUtil.h:254
Scalar * m_data
Definition: BlasUtil.h:264
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void storePacket(Index i, const PacketType &p) const
Definition: BlasUtil.h:259
const internal::variable_if_dynamic< Index, Incr > m_incr
Definition: BlasUtil.h:265
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Scalar & operator()(Index i) const
Definition: BlasUtil.h:249
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE BlasLinearMapper(Scalar *data, Index incr)
Definition: BlasUtil.h:243
Definition: BlasUtil.h:53
Scalar * m_data
Definition: BlasUtil.h:71
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Packet load(Index i) const
Definition: BlasUtil.h:61
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Scalar operator()(Index i) const
Definition: BlasUtil.h:57
EIGEN_DEVICE_FUNC bool aligned(Index i) const
Definition: BlasUtil.h:66
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE BlasVectorMapper(Scalar *data)
Definition: BlasUtil.h:55
BlasVectorMapper< Scalar, Index > VectorMapper
Definition: BlasUtil.h:168
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE LinearMapper getLinearMapper(Index i, Index j) const
Definition: BlasUtil.h:182
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketType loadPacket(Index i, Index j) const
Definition: BlasUtil.h:197
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void storePacketBlock(Index i, Index j, const PacketBlock< SubPacket, n > &block) const
Definition: BlasUtil.h:227
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE SubPacket gatherPacket(Index i, Index j) const
Definition: BlasUtil.h:212
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Scalar & operator()(Index i, Index j) const
Definition: BlasUtil.h:192
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE blas_data_mapper< Scalar, Index, StorageOrder, AlignmentType > getSubMapper(Index i, Index j) const
Definition: BlasUtil.h:178
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE blas_data_mapper(Scalar *data, Index stride, Index incr=1)
Definition: BlasUtil.h:170
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE VectorMapper getVectorMapper(Index i, Index j) const
Definition: BlasUtil.h:186
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void scatterPacket(Index i, Index j, const SubPacket &p) const
Definition: BlasUtil.h:207
EIGEN_DEVICE_FUNC const Scalar * data() const
Definition: BlasUtil.h:217
EIGEN_DEVICE_FUNC Index firstAligned(Index size) const
Definition: BlasUtil.h:219
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketT load(Index i, Index j) const
Definition: BlasUtil.h:202
EIGEN_DEVICE_FUNC const Index stride() const
Definition: BlasUtil.h:216
BlasLinearMapper< Scalar, Index, AlignmentType > LinearMapper
Definition: BlasUtil.h:167
Definition: BlasUtil.h:270
const internal::variable_if_dynamic< Index, Incr > m_incr
Definition: BlasUtil.h:384
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketType loadPacket(Index i, Index j) const
Definition: BlasUtil.h:291
Scalar *EIGEN_RESTRICT m_data
Definition: BlasUtil.h:382
const Index m_stride
Definition: BlasUtil.h:383
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE LinearMapper getLinearMapper(Index i, Index j) const
Definition: BlasUtil.h:281
BlasLinearMapper< Scalar, Index, AlignmentType, Incr > LinearMapper
Definition: BlasUtil.h:272
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketT load(Index i, Index j) const
Definition: BlasUtil.h:296
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE blas_data_mapper(Scalar *data, Index stride, Index incr)
Definition: BlasUtil.h:274
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void storePacketBlock(Index i, Index j, const PacketBlock< SubPacket, n > &block) const
Definition: BlasUtil.h:377
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE blas_data_mapper getSubMapper(Index i, Index j) const
Definition: BlasUtil.h:277
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE SubPacket gatherPacket(Index i, Index j) const
Definition: BlasUtil.h:306
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Scalar & operator()(Index i, Index j) const
Definition: BlasUtil.h:286
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void scatterPacket(Index i, Index j, const SubPacket &p) const
Definition: BlasUtil.h:301
Definition: BlasUtil.h:389
EIGEN_ALWAYS_INLINE const_blas_data_mapper< Scalar, Index, StorageOrder > getSubMapper(Index i, Index j) const
Definition: BlasUtil.h:393
EIGEN_ALWAYS_INLINE const_blas_data_mapper(const Scalar *data, Index stride)
Definition: BlasUtil.h:391
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE EIGEN_CONSTEXPR T value()
Definition: XprHelper.h:135
type
Definition: core.h:575
AlignmentType
Enum for indicating whether a buffer is aligned or not.
Definition: Constants.h:232
@ RowMajor
Storage order is row major (see TopicStorageOrders).
Definition: Constants.h:321
const unsigned int DirectAccessBit
Means that the underlying array of coefficients can be directly accessed as a plain strided array.
Definition: Constants.h:155
std::size_t UIntPtr
Definition: Meta.h:92
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE ResScalar combine_scalar_factors(const ResScalar &alpha, const Lhs &lhs, const Rhs &rhs)
Definition: BlasUtil.h:568
EIGEN_DEVICE_FUNC void prefetch(const Scalar *addr)
Definition: GenericPacketMath.h:719
EIGEN_CONSTEXPR Index size(const T &x)
Definition: Meta.h:479
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE const T::Scalar * extract_data(const T &m)
Definition: BlasUtil.h:533
EIGEN_DEVICE_FUNC Index first_default_aligned(const Scalar *array, Index size)
Definition: Memory.h:497
Namespace containing all symbols from the Eigen library.
Definition: MatrixExponential.h:16
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:74
@ Specialized
Definition: Constants.h:310
Definition: Eigen_Colamd.h:50
Definition: StdDeque.h:50
Holds information about the various numeric (i.e.
Definition: NumTraits.h:233
Definition: GenericPacketMath.h:1014
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(Scalar *to, const Index stride, Index i, Index j, const PacketBlock< Packet, n > &block) const
Definition: BlasUtil.h:154
PacketBlockManagement< Index, Scalar, Packet, n, idx - 1, RowMajor > pbm
Definition: BlasUtil.h:132
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(Scalar *to, const Index stride, Index i, Index j, const PacketBlock< Packet, n > &block) const
Definition: BlasUtil.h:133
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(Scalar *to, const Index stride, Index i, Index j, const PacketBlock< Packet, n > &block) const
Definition: BlasUtil.h:142
Definition: BlasUtil.h:120
PacketBlockManagement< Index, Scalar, Packet, n, idx - 1, StorageOrder > pbm
Definition: BlasUtil.h:121
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(Scalar *to, const Index stride, Index i, Index j, const PacketBlock< Packet, n > &block) const
Definition: BlasUtil.h:122
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(const blas_data_mapper< Scalar, Index, StorageOrder, AlignmentType, Incr > *, Index, Index, const PacketBlock< SubPacket, n > &) const
Definition: BlasUtil.h:358
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(const blas_data_mapper< Scalar, Index, StorageOrder, AlignmentType, Incr > *sup, Index i, Index j, const PacketBlock< SubPacket, n > &block) const
Definition: BlasUtil.h:344
storePacketBlock_helper< SubPacket, std::complex< double >, n, idx-1 > spbh
Definition: BlasUtil.h:343
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(const blas_data_mapper< Scalar, Index, StorageOrder, AlignmentType, Incr > *, Index, Index, const PacketBlock< SubPacket, n > &) const
Definition: BlasUtil.h:372
storePacketBlock_helper< SubPacket, std::complex< float >, n, idx-1 > spbh
Definition: BlasUtil.h:328
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(const blas_data_mapper< Scalar, Index, StorageOrder, AlignmentType, Incr > *sup, Index i, Index j, const PacketBlock< SubPacket, n > &block) const
Definition: BlasUtil.h:329
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(const blas_data_mapper< Scalar, Index, StorageOrder, AlignmentType, Incr > *, Index, Index, const PacketBlock< SubPacket, n > &) const
Definition: BlasUtil.h:365
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void store(const blas_data_mapper< Scalar, Index, StorageOrder, AlignmentType, Incr > *sup, Index i, Index j, const PacketBlock< SubPacket, n > &block) const
Definition: BlasUtil.h:315
storePacketBlock_helper< SubPacket, ScalarT, n, idx-1 > spbh
Definition: BlasUtil.h:314
CwiseBinaryOp< scalar_product_op< Scalar >, const CwiseNullaryOp< scalar_constant_op< Scalar >, Plain >, NestedXpr > XprType
Definition: BlasUtil.h:451
CwiseBinaryOp< scalar_product_op< Scalar >, NestedXpr, const CwiseNullaryOp< scalar_constant_op< Scalar >, Plain > > XprType
Definition: BlasUtil.h:465
static ExtractType extract(const XprType &x)
Definition: BlasUtil.h:438
CwiseUnaryOp< scalar_conjugate_op< Scalar >, NestedXpr > XprType
Definition: BlasUtil.h:431
static Scalar extractScalarFactor(const XprType &x)
Definition: BlasUtil.h:439
static ExtractType extract(const XprType &x)
Definition: BlasUtil.h:488
CwiseUnaryOp< scalar_opposite_op< Scalar >, NestedXpr > XprType
Definition: BlasUtil.h:486
static Scalar extractScalarFactor(const XprType &x)
Definition: BlasUtil.h:489
static ExtractType extract(const XprType &x)
Definition: BlasUtil.h:510
Transpose< NestedXpr > XprType
Definition: BlasUtil.h:500
blas_traits< NestedXpr > Base
Definition: BlasUtil.h:499
NestedXpr::Scalar Scalar
Definition: BlasUtil.h:498
conditional< bool(Base::HasUsableDirectAccess), ExtractType, typenameExtractType::PlainObject >::type DirectLinearAccessType
Definition: BlasUtil.h:506
Transpose< const typename Base::_ExtractType > _ExtractType
Definition: BlasUtil.h:502
static Scalar extractScalarFactor(const XprType &x)
Definition: BlasUtil.h:511
Transpose< const typename Base::_ExtractType > ExtractType
Definition: BlasUtil.h:501
Definition: BlasUtil.h:403
conditional< bool(HasUsableDirectAccess), ExtractType, typename_ExtractType::PlainObject >::type DirectLinearAccessType
Definition: BlasUtil.h:420
static EIGEN_DEVICE_FUNC const Scalar extractScalarFactor(const XprType &)
Definition: BlasUtil.h:422
static EIGEN_DEVICE_FUNC ExtractType extract(const XprType &x)
Definition: BlasUtil.h:421
const XprType & ExtractType
Definition: BlasUtil.h:405
@ HasUsableDirectAccess
Definition: BlasUtil.h:411
@ NeedToConjugate
Definition: BlasUtil.h:410
@ HasScalarFactor
Definition: BlasUtil.h:415
@ IsComplex
Definition: BlasUtil.h:408
@ IsTransposed
Definition: BlasUtil.h:409
traits< XprType >::Scalar Scalar
Definition: BlasUtil.h:404
XprType _ExtractType
Definition: BlasUtil.h:406
EIGEN_DEVICE_FUNC static EIGEN_ALWAYS_INLINE bool run(const Lhs &lhs, const Rhs &rhs)
Definition: BlasUtil.h:557
EIGEN_DEVICE_FUNC static EIGEN_ALWAYS_INLINE bool run(const bool &alpha, const Lhs &lhs, const Rhs &rhs)
Definition: BlasUtil.h:561
combine_scalar_factors extracts and multiplies factors from GEMM and GEMV products.
Definition: BlasUtil.h:544
EIGEN_DEVICE_FUNC static EIGEN_ALWAYS_INLINE ResScalar run(const Lhs &lhs, const Rhs &rhs)
Definition: BlasUtil.h:545
EIGEN_DEVICE_FUNC static EIGEN_ALWAYS_INLINE ResScalar run(const ResScalar &alpha, const Lhs &lhs, const Rhs &rhs)
Definition: BlasUtil.h:549
Definition: Meta.h:109
static T::Scalar * run(const T &)
Definition: BlasUtil.h:529
Definition: BlasUtil.h:520
EIGEN_DEVICE_FUNC static EIGEN_ALWAYS_INLINE const T::Scalar * run(const T &m)
Definition: BlasUtil.h:521
Definition: BlasUtil.h:28
Definition: BlasUtil.h:25
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE NumTraits< Scalar >::Real run(const Scalar &x)
Definition: BlasUtil.h:48
Definition: BlasUtil.h:42
static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE To run(const From &x)
Definition: BlasUtil.h:43
Definition: DenseCoeffsBase.h:659
Definition: UnaryFunctors.h:109
Definition: NullaryFunctors.h:18
Definition: UnaryFunctors.h:22
Definition: BinaryFunctors.h:71
Definition: ForwardDeclarations.h:17
Definition: format.h:1544