devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge.

Browse Source
This commit is contained in:
Rasmus Munk Larsen
2016-05-18 15:16:06 -07:00
parent 7df811cfe5 f519fca72b
commit 5624219b6b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -160,6 +160,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable> {
#else #else
size_t num_threads = device.numThreads(); size_t num_threads = device.numThreads();
if (num_threads > 1) { if (num_threads > 1) {
cost = evaluator.costPerCoeff(Vectorizable)
num_threads = TensorCostModel<ThreadPoolDevice>::numThreads( num_threads = TensorCostModel<ThreadPoolDevice>::numThreads(
size, evaluator.costPerCoeff(Vectorizable), num_threads); size, evaluator.costPerCoeff(Vectorizable), num_threads);
} }
@@ -183,7 +184,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable> {
} }
barrier.Wait(); barrier.Wait();
} }
#endif #endif // defined(!EIGEN_USE_SIMPLE_THREAD_POOL)
} }
evaluator.cleanup(); evaluator.cleanup();
} }