diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2018-09-14 16:55:05 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2018-09-14 16:55:05 -0700 |
| commit | 48633757235b522b37e60022dfbe8813c6cdfdc2 (patch) | |
| tree | 09a63ff38bf2409f7042ec3a00ce29cdd18324f2 /unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | |
| parent | 281e631839436221f8fdd3b18334c866070399dd (diff) | |
Explicitly construct tensor block dimensions from evaluator dimensions
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h index ba5ab1396..bfe1f97b8 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h @@ -282,8 +282,9 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable, /*Tileable*/ tr TensorOpCost cost = evaluator.costPerCoeff(Vectorizable); double taskSize = TensorCostModel<ThreadPoolDevice>::taskSize(1, cost); size_t block_size = static_cast<size_t>(1.0 / taskSize); - TensorBlockMapper block_mapper(evaluator.dimensions(), block_shape, - block_size); + TensorBlockMapper block_mapper( + typename TensorBlockMapper::Dimensions(evaluator.dimensions()), + block_shape, block_size); block_size = block_mapper.block_dims_total_size(); const size_t aligned_blocksize = EIGEN_MAX_ALIGN_BYTES * |