GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-09-14 18:33:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Specialize ThreadPoolDevice::enqueueNotification for the case with no args. As an example this reduces binary size of an TensorFlow demo app for Android by about 2.5%.

Browse Source
This commit is contained in:
Rasmus Munk Larsen 2017-10-13 15:58:12 -07:00
parent 1b7294f6fc
commit f349507e02
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -154,7 +154,11 @@ struct ThreadPoolDevice {
template <class Function, class... Args> template <class Function, class... Args>
EIGEN_STRONG_INLINE void enqueueNoNotification(Function&& f, Args&&... args) const { EIGEN_STRONG_INLINE void enqueueNoNotification(Function&& f, Args&&... args) const {
pool_->Schedule(std::bind(f, args...)); if (sizeof...(args) > 0) {
pool_->Schedule(std::bind(f, args...));
} else {
pool_->Schedule(f);
}
} }
// Returns a logical thread index between 0 and pool_->NumThreads() - 1 if // Returns a logical thread index between 0 and pool_->NumThreads() - 1 if