mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-08-12 03:39:01 +08:00
Address comments by bsteiner.
This commit is contained in:
parent
e55deb21c5
commit
989e419328
@ -10,9 +10,9 @@
|
||||
#ifndef EIGEN_CXX11_TENSOR_TENSOR_COST_MODEL_H
|
||||
#define EIGEN_CXX11_TENSOR_TENSOR_COST_MODEL_H
|
||||
|
||||
//#if !defined(EIGEN_USE_GPU)
|
||||
//#define EIGEN_USE_COST_MODEL
|
||||
//#endif
|
||||
#if !defined(EIGEN_USE_GPU)
|
||||
#define EIGEN_USE_COST_MODEL
|
||||
#endif
|
||||
|
||||
namespace Eigen {
|
||||
|
||||
|
@ -260,7 +260,7 @@ struct ThreadPoolDevice {
|
||||
barrier.Wait();
|
||||
}
|
||||
|
||||
// Convinience wrapper for parallelFor that does not align blocks.
|
||||
// Convenience wrapper for parallelFor that does not align blocks.
|
||||
void parallelFor(Index n, const TensorOpCost& cost,
|
||||
std::function<void(Index, Index)> f) const {
|
||||
parallelFor(n, cost, nullptr, std::move(f));
|
||||
|
@ -170,7 +170,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable> {
|
||||
}
|
||||
barrier.Wait();
|
||||
}
|
||||
#endif // EIGEN_USE_NONBLOCKING_THREAD_POOL
|
||||
#endif // defined(EIGEN_USE_NONBLOCKING_THREAD_POOL) && defined(EIGEN_USE_COST_MODEL)
|
||||
}
|
||||
evaluator.cleanup();
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user