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

Move from rvalue arguments in ThreadPool enqueue* methods

Browse Source
This commit is contained in:
Eugene Zhulenev 2018-10-16 16:48:32 -07:00
parent 217d839816
commit 9e96e91936
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -109,26 +109,30 @@ struct ThreadPoolDevice {
} }
template <class Function, class... Args> template <class Function, class... Args>
EIGEN_STRONG_INLINE Notification* enqueue(Function&& f, Args&&... args) const { EIGEN_STRONG_INLINE Notification* enqueue(Function&& f,
Args&&... args) const {
Notification* n = new Notification(); Notification* n = new Notification();
pool_->Schedule(std::bind(&FunctionWrapperWithNotification<Function, Args...>::run, n, f, args...)); pool_->Schedule(
std::bind(&FunctionWrapperWithNotification<Function, Args...>::run, n,
std::move(f), args...));
return n; return n;
} }
template <class Function, class... Args> template <class Function, class... Args>
EIGEN_STRONG_INLINE void enqueue_with_barrier(Barrier* b, EIGEN_STRONG_INLINE void enqueue_with_barrier(Barrier* b, Function&& f,
Function&& f,
Args&&... args) const { Args&&... args) const {
pool_->Schedule(std::bind( pool_->Schedule(
&FunctionWrapperWithBarrier<Function, Args...>::run, b, f, args...)); std::bind(&FunctionWrapperWithBarrier<Function, Args...>::run, b,
std::move(f), args...));
} }
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 {
if (sizeof...(args) > 0) { if (sizeof...(args) > 0) {
pool_->Schedule(std::bind(f, args...)); pool_->Schedule(std::bind(std::move(f), args...));
} else { } else {
pool_->Schedule(f); pool_->Schedule(std::move(f));
} }
} }