diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2019-08-30 15:13:38 -0700 |
---|---|---|
committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-08-30 15:13:38 -0700 |
commit | f0b36fb9a405400e82b73ea70097b8ae3cd1095a (patch) | |
tree | d3a2903422799257720d2d4989bcd845ab2ae27e /unsupported/Eigen/CXX11/src/ThreadPool | |
parent | 619cea94916e7531a839ee0ff657714857921db8 (diff) |
evalSubExprsIfNeededAsync + async TensorContractionThreadPool
Diffstat (limited to 'unsupported/Eigen/CXX11/src/ThreadPool')
-rw-r--r-- | unsupported/Eigen/CXX11/src/ThreadPool/Barrier.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/ThreadPool/Barrier.h b/unsupported/Eigen/CXX11/src/ThreadPool/Barrier.h index bae68e1fb..e4c59dc3d 100644 --- a/unsupported/Eigen/CXX11/src/ThreadPool/Barrier.h +++ b/unsupported/Eigen/CXX11/src/ThreadPool/Barrier.h @@ -25,6 +25,9 @@ class Barrier { void Notify() { unsigned int v = state_.fetch_sub(2, std::memory_order_acq_rel) - 2; if (v != 1) { + // Clear the lowest bit (waiter flag) and check that the original state + // value was not zero. If it was zero, it means that notify was called + // more times than the original count. eigen_plain_assert(((v + 2) & ~1) != 0); return; // either count has not dropped to 0, or waiter is not waiting } |