GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-13 20:26:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Address comments by bsteiner.

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
unsupported/Eigen/CXX11/src/Tensor/TensorCostModel.h
View File

@ -10,9 +10,9 @@
#ifndef EIGEN_CXX11_TENSOR_TENSOR_COST_MODEL_H #ifndef EIGEN_CXX11_TENSOR_TENSOR_COST_MODEL_H
#define EIGEN_CXX11_TENSOR_TENSOR_COST_MODEL_H #define EIGEN_CXX11_TENSOR_TENSOR_COST_MODEL_H
//#if !defined(EIGEN_USE_GPU) #if !defined(EIGEN_USE_GPU)
//#define EIGEN_USE_COST_MODEL #define EIGEN_USE_COST_MODEL
//#endif #endif
namespace Eigen { namespace Eigen {

2
unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h
View File

@ -260,7 +260,7 @@ struct ThreadPoolDevice {
barrier.Wait(); 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, void parallelFor(Index n, const TensorOpCost& cost,
std::function<void(Index, Index)> f) const { std::function<void(Index, Index)> f) const {
parallelFor(n, cost, nullptr, std::move(f)); parallelFor(n, cost, nullptr, std::move(f));

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

@ -170,7 +170,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable> {
} }
barrier.Wait(); barrier.Wait();
} }
#endif // EIGEN_USE_NONBLOCKING_THREAD_POOL #endif // defined(EIGEN_USE_NONBLOCKING_THREAD_POOL) && defined(EIGEN_USE_COST_MODEL)
} }
evaluator.cleanup(); evaluator.cleanup();
} }