diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2019-10-14 14:31:59 -0700 |
---|---|---|
committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-10-14 14:31:59 -0700 |
commit | d380c23b2cc0b02e10819e779c73cde2c62603b2 (patch) | |
tree | 09d204c87ed6a9f55aa0c6305d7e4199a71dbf8a /unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | |
parent | 39fb9eeccf2e79542acad9bbf5196e462c1b2cee (diff) |
Block evaluation for TensorGenerator/TensorReverse/TensorShuffling
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | 13 |
1 files changed, 0 insertions, 13 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h index 6ad6327a6..97ac96db1 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h @@ -521,19 +521,6 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable, static EIGEN_STRONG_INLINE void run(const Expression& expr, const ThreadPoolDevice& device) { Evaluator evaluator(expr, device); - Index total_size = array_prod(evaluator.dimensions()); - Index cache_size = device.firstLevelCacheSize() / sizeof(Scalar); - - // TODO(ezuhulenev): For small expressions cost of block mapping and - // resource requirements gathering dominates the cost of expression - // evaluatiuon. - if (total_size < cache_size && - !ExpressionHasTensorBroadcastingOp<Expression>::value) { - internal::TensorExecutor<Expression, ThreadPoolDevice, Vectorizable, - /*Tiling=*/TiledEvaluation::Off>::run(expr, device); - evaluator.cleanup(); - return; - } const bool needs_assign = evaluator.evalSubExprsIfNeeded(nullptr); if (needs_assign) { |