diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2018-08-09 09:44:07 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2018-08-09 09:44:07 -0700 |
| commit | cfaedb38cd662def3b5684a20965b3bc1b0d6a3f (patch) | |
| tree | 82dc639a1f91a2eebd9d1afb777945dd8830ff70 /unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | |
| parent | 1c8b9e10a791cb43b4f730dcb5d7889099cc1c68 (diff) | |
Fix bug in a test + compilation errors
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h index f11241f83..4bc18bd97 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h @@ -115,6 +115,7 @@ class TensorExecutor<Expression, DefaultDevice, Vectorizable, const DefaultDevice& device = DefaultDevice()) { typedef TensorBlock<ScalarNoConst, StorageIndex, NumDims, Evaluator::Layout> TensorBlock; + typedef typename TensorBlock::Dimensions TensorBlockDimensions; typedef TensorBlockMapper<ScalarNoConst, StorageIndex, NumDims, Evaluator::Layout> TensorBlockMapper; @@ -141,8 +142,9 @@ class TensorExecutor<Expression, DefaultDevice, Vectorizable, evaluator.getResourceRequirements(&resources); MergeResourceRequirements(resources, &block_shape, &block_total_size); - TensorBlockMapper block_mapper(evaluator.dimensions(), block_shape, - block_total_size); + TensorBlockMapper block_mapper( + TensorBlockDimensions(evaluator.dimensions()), block_shape, + block_total_size); block_total_size = block_mapper.block_dims_total_size(); Scalar* data = static_cast<Scalar*>( |