GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 03:39:01 +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
commit 5624219b6b
1 changed files with 2 additions and 1 deletions
Show Stats 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();
} }