diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2014-09-03 11:38:13 -0700 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2014-09-03 11:38:13 -0700 |
| commit | b24fe22b1a4518f27ca064d496bfdb6c96d973ab (patch) | |
| tree | 78a99bc733484a82c430d7724bdfafe026630a0e /unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h | |
| parent | 2959045f2fe111f93b23517fd6f7afe49720a290 (diff) | |
Improved the performance of the tensor convolution code by a factor of about 4.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h b/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h index db716a80e..587cbd5ca 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorEvalTo.h @@ -108,8 +108,9 @@ struct TensorEvaluator<const TensorEvalToOp<ArgType>, Device> EIGEN_DEVICE_FUNC const Dimensions& dimensions() const { return m_impl.dimensions(); } - EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void evalSubExprsIfNeeded() { - m_impl.evalSubExprsIfNeeded(); + EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool evalSubExprsIfNeeded(Scalar*) { + m_impl.evalSubExprsIfNeeded(NULL); + return true; } EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void evalScalar(Index i) { @@ -134,6 +135,8 @@ struct TensorEvaluator<const TensorEvalToOp<ArgType>, Device> return internal::ploadt<Packet, LoadMode>(m_buffer + index); } + Scalar* data() const { return NULL; } + private: TensorEvaluator<ArgType, Device> m_impl; const Device& m_device; |