GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-06-04 18:54:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enable direct access for NestByValue.

Browse Source
This commit is contained in:
Antonio Sánchez 2022-12-07 18:21:45 +00:00 committed by Rasmus Munk Larsen
parent b59f18b4f7
commit 03c9b4738c
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
Eigen/src/Core/NestByValue.h
View File

@ -43,6 +43,8 @@ template<typename ExpressionType> class NestByValue
public: public:
typedef typename internal::dense_xpr_base<NestByValue>::type Base; typedef typename internal::dense_xpr_base<NestByValue>::type Base;
static constexpr bool HasDirectAccess = internal::has_direct_access<ExpressionType>::ret;
EIGEN_DENSE_PUBLIC_INTERFACE(NestByValue) EIGEN_DENSE_PUBLIC_INTERFACE(NestByValue)
EIGEN_DEVICE_FUNC explicit inline NestByValue(const ExpressionType& matrix) : m_expression(matrix) {} EIGEN_DEVICE_FUNC explicit inline NestByValue(const ExpressionType& matrix) : m_expression(matrix) {}
@ -54,6 +56,18 @@ template<typename ExpressionType> class NestByValue
EIGEN_DEVICE_FUNC const ExpressionType& nestedExpression() const { return m_expression; } EIGEN_DEVICE_FUNC const ExpressionType& nestedExpression() const { return m_expression; }
EIGEN_DEVICE_FUNC typename std::enable_if<HasDirectAccess, const Scalar*>::type data() const {
return m_expression.data();
}
EIGEN_DEVICE_FUNC typename std::enable_if<HasDirectAccess, Index>::type innerStride() const {
return m_expression.innerStride();
}
EIGEN_DEVICE_FUNC typename std::enable_if<HasDirectAccess, Index>::type outerStride() const {
return m_expression.outerStride();
}
protected: protected:
const ExpressionType m_expression; const ExpressionType m_expression;
}; };

2
test/nestbyvalue.cpp
View File

@ -33,5 +33,7 @@ EIGEN_DECLARE_TEST(nestbyvalue)
MatrixXd b = x; MatrixXd b = x;
VERIFY_IS_EQUAL(nb_temporaries,6+1); VERIFY_IS_EQUAL(nb_temporaries,6+1);
VERIFY_IS_APPROX(b, a.rowwise().reverse().eval() + (a+a).eval()); VERIFY_IS_APPROX(b, a.rowwise().reverse().eval() + (a+a).eval());
// Block expressions work with dense NestByValue.
VERIFY_IS_APPROX(b, a.nestByValue().rowwise().reverse().eval() + (a.nestByValue()+a.nestByValue()).eval());
} }
} }