diff options
author | Antonio Sanchez <cantonios@google.com> | 2021-01-06 09:13:28 -0800 |
---|---|---|
committer | Antonio Sanchez <cantonios@google.com> | 2021-01-06 09:16:52 -0800 |
commit | 166fcdecdb9178fe7f3eea38f67fb3f5848a4ae2 (patch) | |
tree | c8594549212aa04213f33be36fb3293679ce153b | |
parent | bb1de9dbdede6669c2c86c028a9deff637e3d1f6 (diff) |
Allow CwiseUnaryView to be used on device.
Added `EIGEN_DEVICE_FUNC` to methods.
-rw-r--r-- | Eigen/src/Core/CwiseUnaryView.h | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/Eigen/src/Core/CwiseUnaryView.h b/Eigen/src/Core/CwiseUnaryView.h index ff3134d43..02b034de7 100644 --- a/Eigen/src/Core/CwiseUnaryView.h +++ b/Eigen/src/Core/CwiseUnaryView.h @@ -64,23 +64,23 @@ class CwiseUnaryView : public CwiseUnaryViewImpl<ViewOp, MatrixType, typename in typedef typename internal::ref_selector<MatrixType>::non_const_type MatrixTypeNested; typedef typename internal::remove_all<MatrixType>::type NestedExpression; - explicit inline CwiseUnaryView(MatrixType& mat, const ViewOp& func = ViewOp()) + explicit EIGEN_DEVICE_FUNC inline CwiseUnaryView(MatrixType& mat, const ViewOp& func = ViewOp()) : m_matrix(mat), m_functor(func) {} EIGEN_INHERIT_ASSIGNMENT_OPERATORS(CwiseUnaryView) - EIGEN_STRONG_INLINE Index rows() const { return m_matrix.rows(); } - EIGEN_STRONG_INLINE Index cols() const { return m_matrix.cols(); } + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index rows() const { return m_matrix.rows(); } + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index cols() const { return m_matrix.cols(); } /** \returns the functor representing unary operation */ - const ViewOp& functor() const { return m_functor; } + EIGEN_DEVICE_FUNC const ViewOp& functor() const { return m_functor; } /** \returns the nested expression */ - const typename internal::remove_all<MatrixTypeNested>::type& + EIGEN_DEVICE_FUNC const typename internal::remove_all<MatrixTypeNested>::type& nestedExpression() const { return m_matrix; } /** \returns the nested expression */ - typename internal::remove_reference<MatrixTypeNested>::type& + EIGEN_DEVICE_FUNC typename internal::remove_reference<MatrixTypeNested>::type& nestedExpression() { return m_matrix; } protected: |