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

Make it possible to override the synchonization primitives used by the threadpool using macros.

Browse Source
This commit is contained in:
Rasmus Munk Larsen
2023-05-09 19:36:17 +00:00
parent 1321821e86
commit 96c42771d6
6 changed files with 27 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Eigen/src/ThreadPool/Barrier.h
View File

@@ -33,7 +33,7 @@ class Barrier {
eigen_plain_assert(((v + 2) & ~1) != 0);
return; // either count has not dropped to 0, or waiter is not waiting
}
std::unique_lock<std::mutex> l(mu_);
EIGEN_MUTEX_LOCK l(mu_);
eigen_plain_assert(!notified_);
notified_ = true;
cv_.notify_all();
@@ -42,15 +42,15 @@ class Barrier {
void Wait() {
unsigned int v = state_.fetch_or(1, std::memory_order_acq_rel);
if ((v >> 1) == 0) return;
std::unique_lock<std::mutex> l(mu_);
EIGEN_MUTEX_LOCK l(mu_);
while (!notified_) {
cv_.wait(l);
}
}
private:
std::mutex mu_;
std::condition_variable cv_;
EIGEN_MUTEX mu_;
EIGEN_CONDVAR cv_;
std::atomic<unsigned int> state_; // low bit is waiter flag
bool notified_;
};