10#ifndef EIGEN_BLASUTIL_H
11#define EIGEN_BLASUTIL_H
21template<
typename LhsScalar,
typename RhsScalar,
typename Index,
typename DataMapper,
int mr,
int nr,
bool ConjugateLhs=false,
bool ConjugateRhs=false>
24template<
typename Scalar,
typename Index,
typename DataMapper,
int nr,
int StorageOrder,
bool Conjugate = false,
bool PanelMode=false>
27template<
typename Scalar,
typename Index,
typename DataMapper,
int Pack1,
int Pack2,
typename Packet,
int StorageOrder,
bool Conjugate = false,
bool PanelMode = false>
32 typename LhsScalar,
int LhsStorageOrder,
bool ConjugateLhs,
33 typename RhsScalar,
int RhsStorageOrder,
bool ConjugateRhs,
34 int ResStorageOrder,
int ResInnerStride>
37template<
typename Index,
38 typename LhsScalar,
typename LhsMapper,
int LhsStorageOrder,
bool ConjugateLhs,
39 typename RhsScalar,
typename RhsMapper,
bool ConjugateRhs,
int Version=Specialized>
43 EIGEN_DEVICE_FUNC
static EIGEN_STRONG_INLINE To run(
const From& x) {
return To(x); }
48 static EIGEN_STRONG_INLINE
typename NumTraits<Scalar>::Real run(
const Scalar& x) {
return numext::real(x); }
52template<
typename Scalar,
typename Index>
55 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
BlasVectorMapper(Scalar *data) : m_data(data) {}
57 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Scalar operator()(
Index i)
const {
60 template <
typename Packet,
int AlignmentType>
61 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
Packet load(
Index i)
const {
62 return ploadt<Packet, AlignmentType>(m_data + i);
65 template <
typename Packet>
66 EIGEN_DEVICE_FUNC
bool aligned(
Index i)
const {
67 return (UIntPtr(m_data+i)%
sizeof(
Packet))==0;
74template<
typename Scalar,
typename Index,
int AlignmentType,
int Incr=1>
77template<
typename Scalar,
typename Index,
int AlignmentType>
84 EIGEN_ONLY_USED_FOR_DEBUG(incr);
85 eigen_assert(incr==1);
88 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
void prefetch(
int i)
const {
89 internal::prefetch(&
operator()(i));
92 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Scalar& operator()(
Index i)
const {
96 template<
typename PacketType>
97 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketType loadPacket(
Index i)
const {
98 return ploadt<PacketType, AlignmentType>(m_data + i);
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);
111template<
typename Scalar,
typename Index,
int StorageOrder,
int AlignmentType = Unaligned,
int Incr = 1>
118template<
typename Index,
typename Scalar,
typename Packet,
int n,
int idx,
int StorageOrder>
123 pbm.store(to, stride, i, j, block);
124 pstoreu<Scalar>(to + i + (j + idx)*stride, block.packet[idx]);
129template<
typename Index,
typename Scalar,
typename Packet,
int n,
int idx>
134 pbm.store(to, stride, i, j, block);
135 pstoreu<Scalar>(to + j + (i + idx)*stride, block.packet[idx]);
139template<
typename Index,
typename Scalar,
typename Packet,
int n,
int StorageOrder>
143 EIGEN_UNUSED_VARIABLE(to);
144 EIGEN_UNUSED_VARIABLE(stride);
145 EIGEN_UNUSED_VARIABLE(i);
146 EIGEN_UNUSED_VARIABLE(j);
147 EIGEN_UNUSED_VARIABLE(block);
151template<
typename Index,
typename Scalar,
typename Packet,
int n>
155 EIGEN_UNUSED_VARIABLE(to);
156 EIGEN_UNUSED_VARIABLE(stride);
157 EIGEN_UNUSED_VARIABLE(i);
158 EIGEN_UNUSED_VARIABLE(j);
159 EIGEN_UNUSED_VARIABLE(block);
163template<
typename Scalar,
typename Index,
int StorageOrder,
int AlignmentType>
171 : m_data(data), m_stride(stride)
173 EIGEN_ONLY_USED_FOR_DEBUG(incr);
174 eigen_assert(incr==1);
192 EIGEN_ALWAYS_INLINE Scalar& operator()(
Index i,
Index j)
const {
193 return m_data[StorageOrder==
RowMajor ? j + i*m_stride : i + j*m_stride];
196 template<
typename PacketType>
197 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketType loadPacket(
Index i,
Index j)
const {
198 return ploadt<PacketType, AlignmentType>(&
operator()(i, j));
201 template <
typename PacketT,
int AlignmentT>
202 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketT load(
Index i,
Index j)
const {
203 return ploadt<PacketT, AlignmentT>(&
operator()(i, j));
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);
211 template<
typename SubPacket>
212 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE SubPacket gatherPacket(
Index i,
Index j)
const {
213 return pgather<Scalar, SubPacket>(&
operator()(i, j), m_stride);
216 EIGEN_DEVICE_FUNC
const Index stride()
const {
return m_stride; }
217 EIGEN_DEVICE_FUNC
const Scalar* data()
const {
return m_data; }
219 EIGEN_DEVICE_FUNC
Index firstAligned(
Index size)
const {
220 if (UIntPtr(m_data)%
sizeof(Scalar)) {
223 return internal::first_default_aligned(m_data, size);
226 template<
typename SubPacket,
int n>
229 pbm.store(m_data, m_stride, i, j, block);
232 Scalar* EIGEN_RESTRICT m_data;
233 const Index m_stride;
239template<
typename Scalar,
typename Index,
int AlignmentType,
int Incr>
243 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
BlasLinearMapper(Scalar *data,
Index incr) : m_data(data), m_incr(incr) {}
245 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
void prefetch(
int i)
const {
246 internal::prefetch(&
operator()(i));
249 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Scalar& operator()(
Index i)
const {
250 return m_data[i*m_incr.value()];
253 template<
typename PacketType>
254 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketType loadPacket(
Index i)
const {
255 return pgather<Scalar,PacketType>(m_data + i*m_incr.value(), m_incr.value());
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());
268template<
typename Scalar,
typename Index,
int StorageOrder,
int AlignmentType,
int Incr>
274 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
blas_data_mapper(Scalar* data,
Index stride,
Index incr) : m_data(data), m_stride(stride), m_incr(incr) {}
286 EIGEN_ALWAYS_INLINE Scalar& operator()(
Index i,
Index j)
const {
287 return m_data[StorageOrder==
RowMajor ? j*m_incr.value() + i*m_stride : i*m_incr.value() + j*m_stride];
290 template<
typename PacketType>
291 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketType loadPacket(
Index i,
Index j)
const {
292 return pgather<Scalar,PacketType>(&
operator()(i, j),m_incr.value());
295 template <
typename PacketT,
int AlignmentT>
296 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE PacketT load(
Index i,
Index j)
const {
297 return pgather<Scalar,PacketT>(&
operator()(i, j),m_incr.value());
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);
305 template<
typename SubPacket>
306 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE SubPacket gatherPacket(
Index i,
Index j)
const {
307 return pgather<Scalar, SubPacket>(&
operator()(i, j), m_stride);
311 template<
typename SubPacket,
typename ScalarT,
int n,
int idx>
316 spbh.store(sup, i,j,block);
317 for(
int l = 0; l < unpacket_traits<SubPacket>::size; l++)
319 ScalarT *v = &sup->operator()(i+l, j+idx);
320 *v = block.packet[idx][l];
325 template<
typename SubPacket,
int n,
int idx>
330 spbh.store(sup,i,j,block);
331 for(
int l = 0; l < unpacket_traits<SubPacket>::size; l++)
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]);
340 template<
typename SubPacket,
int n,
int idx>
345 spbh.store(sup,i,j,block);
346 for(
int l = 0; l < unpacket_traits<SubPacket>::size; l++)
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]);
355 template<
typename SubPacket,
typename ScalarT,
int n>
362 template<
typename SubPacket,
int n>
369 template<
typename SubPacket,
int n>
376 template<
typename SubPacket,
int n>
379 spb.store(
this, i,j,block);
382 Scalar* EIGEN_RESTRICT m_data;
383 const Index m_stride;
388template<
typename Scalar,
typename Index,
int StorageOrder>
405 typedef const XprType& ExtractType;
406 typedef XprType _ExtractType;
409 IsTransposed =
false,
410 NeedToConjugate =
false,
412 && (
bool(XprType::IsVectorAtCompileTime)
415 HasScalarFactor =
false
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); }
426template<
typename Scalar,
typename NestedXpr>
432 typedef typename Base::ExtractType ExtractType;
436 NeedToConjugate = Base::NeedToConjugate ? 0 : IsComplex
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())); }
443template<
typename Scalar,
typename NestedXpr,
typename Plain>
448 HasScalarFactor =
true
452 typedef typename Base::ExtractType ExtractType;
453 static inline EIGEN_DEVICE_FUNC ExtractType extract(
const XprType& x) {
return Base::extract(x.rhs()); }
454 static inline EIGEN_DEVICE_FUNC Scalar extractScalarFactor(
const XprType& x)
455 {
return x.lhs().functor().m_other * Base::extractScalarFactor(x.rhs()); }
457template<
typename Scalar,
typename NestedXpr,
typename Plain>
462 HasScalarFactor =
true
466 typedef typename Base::ExtractType ExtractType;
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; }
471template<
typename Scalar,
typename Plain1,
typename Plain2>
474 :
blas_traits<CwiseNullaryOp<scalar_constant_op<Scalar>,Plain1> >
478template<
typename Scalar,
typename NestedXpr>
483 HasScalarFactor =
true
487 typedef typename Base::ExtractType ExtractType;
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()); }
494template<
typename NestedXpr>
498 typedef typename NestedXpr::Scalar Scalar;
503 typedef typename conditional<bool(Base::HasUsableDirectAccess),
505 typename ExtractType::PlainObject
508 IsTransposed = Base::IsTransposed ? 0 : 1
511 static inline Scalar extractScalarFactor(
const XprType& x) {
return Base::extractScalarFactor(x.nestedExpression()); }
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)
529 static typename T::Scalar* run(
const T&) {
return 0; }
533EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
const typename T::Scalar* extract_data(
const T& m)
542template<
typename ResScalar,
typename Lhs,
typename Rhs>
545 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
static ResScalar run(
const Lhs& lhs,
const Rhs& rhs)
549 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
static ResScalar run(
const ResScalar& alpha,
const Lhs& lhs,
const Rhs& rhs)
554template<
typename Lhs,
typename Rhs>
557 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
static bool run(
const Lhs& lhs,
const Rhs& rhs)
561 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
static bool run(
const bool& alpha,
const Lhs& lhs,
const Rhs& rhs)
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)
572template<
typename ResScalar,
typename Lhs,
typename Rhs>
573EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE ResScalar combine_scalar_factors(
const Lhs& lhs,
const Rhs& rhs)
575 return combine_scalar_factors_impl<ResScalar,Lhs,Rhs>::run(lhs, rhs);
Generic expression where a coefficient-wise binary operator is applied to two expressions.
Definition CwiseBinaryOp.h:84
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
Expression of the transpose of a matrix.
Definition Transpose.h:54
Definition BlasUtil.h:241
Definition BlasUtil.h:270
Definition BlasUtil.h:389
Definition XprHelper.h:130
AlignmentType
Definition Constants.h:232
@ RowMajor
Definition Constants.h:321
const unsigned int DirectAccessBit
Definition Constants.h:155
Namespace containing all symbols from the Eigen library.
Definition common.h:81
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition Meta.h:74
Holds information about the various numeric (i.e. scalar) types allowed by Eigen.
Definition NumTraits.h:233
Definition BlasUtil.h:120
Definition GenericPacketMath.h:1014
Definition BlasUtil.h:313
Definition BlasUtil.h:403
Definition BlasUtil.h:544
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 datatypes.h:12
Definition PacketMath.h:47