GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-06-04 18:54:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Allocate non-const scalar buffer for block evaluation with DefaultDevice

Browse Source
This commit is contained in:
Eugene Zhulenev 2019-07-01 10:55:19 -07:00
parent ffaf658ecd
commit 4ac93f8edc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
unsupported/Eigen/CXX11/src/Tensor/TensorExecutor.h
View File

@ -189,7 +189,7 @@ class TensorExecutor<Expression, DefaultDevice, Vectorizable,
block_total_size); block_total_size);
block_total_size = block_mapper.block_dims_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))); device.allocate(block_total_size * sizeof(Scalar)));
const StorageIndex total_block_count = block_mapper.total_block_count(); const StorageIndex total_block_count = block_mapper.total_block_count();