GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-05-01 16:24:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reduce number of allocations in TensorContractionThreadPool.

Browse Source
This commit is contained in:
Eugene Zhulenev 2018-07-16 14:26:39 -07:00
parent 7ccb623746
commit 6e654f3379
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h
View File

@ -598,11 +598,13 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT
else else
pack_lhs(start, k); pack_lhs(start, k);
} else { } else {
while (end - start > 1) {
Index mid = (start + end) / 2; Index mid = (start + end) / 2;
device_.enqueueNoNotification( device_.enqueueNoNotification(
[=]() { enqueue_packing_helper(mid, end, k, rhs); }); [=]() { enqueue_packing_helper(mid, end, k, rhs); });
device_.enqueueNoNotification( end = mid;
[=]() { enqueue_packing_helper(start, mid, k, rhs); }); }
enqueue_packing_helper(start, end, k, rhs);
} }
} }