devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +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
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>
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