GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-05-01 08:14:10 +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

12
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 {
Index mid = (start + end) / 2; while (end - start > 1) {
device_.enqueueNoNotification( Index mid = (start + end) / 2;
[=]() { enqueue_packing_helper(mid, end, k, rhs); }); device_.enqueueNoNotification(
device_.enqueueNoNotification( [=]() { enqueue_packing_helper(mid, end, k, rhs); });
[=]() { enqueue_packing_helper(start, mid, k, rhs); }); end = mid;
}
enqueue_packing_helper(start, end, k, rhs);
} }
} }