Address comments by bsteiner.

This commit is contained in:
Rasmus Munk Larsen 2016-05-12 16:54:19 -07:00
parent e55deb21c5
commit 989e419328
3 changed files with 5 additions and 5 deletions

View File

@ -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 {

View File

@ -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));

View File

@ -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();
}