diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2019-07-01 10:55:19 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-07-01 10:55:19 -0700 |
| commit | 4ac93f8edc09ac051dec73e09f204dd34bf7ebd0 (patch) | |
| tree | 9803b7b27d27dce4fef1388e5dfbbfc7f9ed3be9 /unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | |
| parent | ffaf658ecd08a8ff7fa976a39cad73fb745a2a10 (diff) | |
Allocate non-const scalar buffer for block evaluation with DefaultDevice
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h index f1ae548f7..fddb90d77 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h @@ -189,7 +189,7 @@ class TensorExecutor<Expression, DefaultDevice, Vectorizable, block_total_size); block_total_size = block_mapper.block_dims_total_size(); - Scalar* data = static_cast<Scalar*>( + ScalarNoConst* data = static_cast<ScalarNoConst*>( device.allocate(block_total_size * sizeof(Scalar))); const StorageIndex total_block_count = block_mapper.total_block_count(); |