GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-04-29 15:24:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Started to implement a portable way to yield.

Browse Source
This commit is contained in:
Benoit Steiner 2016-04-19 17:59:58 -07:00
parent 1d0238375d
commit c7c2054bb5
3 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
unsupported/Eigen/CXX11/ThreadPool
View File

@ -45,9 +45,10 @@
#include <functional> #include <functional>
#include <memory> #include <memory>
#include "src/ThreadPool/ThreadLocal.h"
#include "src/ThreadPool/ThreadYield.h"
#include "src/ThreadPool/EventCount.h" #include "src/ThreadPool/EventCount.h"
#include "src/ThreadPool/RunQueue.h" #include "src/ThreadPool/RunQueue.h"
#include "src/ThreadPool/ThreadLocal.h"
#include "src/ThreadPool/ThreadPoolInterface.h" #include "src/ThreadPool/ThreadPoolInterface.h"
#include "src/ThreadPool/ThreadEnvironment.h" #include "src/ThreadPool/ThreadEnvironment.h"
#include "src/ThreadPool/SimpleThreadPool.h" #include "src/ThreadPool/SimpleThreadPool.h"

4
unsupported/Eigen/CXX11/src/ThreadPool/EventCount.h
View File

@ -81,7 +81,7 @@ class EventCount {
if (int64_t((state & kEpochMask) - epoch) < 0) { if (int64_t((state & kEpochMask) - epoch) < 0) {
// The preceeding waiter has not decided on its fate. Wait until it // The preceeding waiter has not decided on its fate. Wait until it
// calls either CancelWait or CommitWait, or is notified. // calls either CancelWait or CommitWait, or is notified.
std::this_thread::yield(); EIGEN_THREAD_YIELD();
state = state_.load(std::memory_order_seq_cst); state = state_.load(std::memory_order_seq_cst);
continue; continue;
} }
@ -112,7 +112,7 @@ class EventCount {
if (int64_t((state & kEpochMask) - epoch) < 0) { if (int64_t((state & kEpochMask) - epoch) < 0) {
// The preceeding waiter has not decided on its fate. Wait until it // The preceeding waiter has not decided on its fate. Wait until it
// calls either CancelWait or CommitWait, or is notified. // calls either CancelWait or CommitWait, or is notified.
std::this_thread::yield(); EIGEN_THREAD_YIELD();
state = state_.load(std::memory_order_relaxed); state = state_.load(std::memory_order_relaxed);
continue; continue;
} }

20
unsupported/Eigen/CXX11/src/ThreadPool/ThreadYield.h Normal file
View File

@ -0,0 +1,20 @@
// This file is part of Eigen, a lightweight C++ template library
// for linear algebra.
//
// Copyright (C) 2016 Benoit Steiner <benoit.steiner.goog@gmail.com>
//
// This Source Code Form is subject to the terms of the Mozilla
// Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
#ifndef EIGEN_CXX11_THREADPOOL_THREAD_YIELD_H
#define EIGEN_CXX11_THREADPOOL_THREAD_YIELD_H
// Try to come up with a portable way to yield
#if EIGEN_COMP_GNUC && EIGEN_GNUC_AT_MOST(4, 7)
#define EIGEN_THREAD_YIELD() sched_yield()
#else
#define EIGEN_THREAD_YIELD() std::this_thread::yield()
#endif
#endif // EIGEN_CXX11_THREADPOOL_THREAD_YIELD_H