mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-08-11 19:29:02 +08:00
Merge.
This commit is contained in:
commit
5624219b6b
@ -160,6 +160,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable> {
|
||||
#else
|
||||
size_t num_threads = device.numThreads();
|
||||
if (num_threads > 1) {
|
||||
cost = evaluator.costPerCoeff(Vectorizable)
|
||||
num_threads = TensorCostModel<ThreadPoolDevice>::numThreads(
|
||||
size, evaluator.costPerCoeff(Vectorizable), num_threads);
|
||||
}
|
||||
@ -183,7 +184,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable> {
|
||||
}
|
||||
barrier.Wait();
|
||||
}
|
||||
#endif
|
||||
#endif // defined(!EIGEN_USE_SIMPLE_THREAD_POOL)
|
||||
}
|
||||
evaluator.cleanup();
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user