Merge branch 'master' of /home/git/concurrency-benchmarks

[c11concurrency-benchmarks.git] / cdschecker_modified_benchmarks / barrier / barrier.h

diff --git a/cdschecker_modified_benchmarks/barrier/barrier.h b/cdschecker_modified_benchmarks/barrier/barrier.h
new file mode 100644 (file)
index 0000000..b52b849
--- /dev/null
+++ b/cdschecker_modified_benchmarks/barrier/barrier.h
@@ -0,0 +1,36 @@
+#include "cds_atomic.h"
+
+class spinning_barrier {
+ public:
+       spinning_barrier (unsigned int n) : n_ (n) {
+               nwait_ = 0;
+               step_ = 0;
+       }
+
+       bool wait() {
+               unsigned int step = step_.load ();
+
+               if (nwait_.fetch_add (1) == n_ - 1) {
+                       /* OK, last thread to come.  */
+                       nwait_.store (0); // XXX: maybe can use relaxed ordering here ??
+                       step_.fetch_add (1, std::memory_order_relaxed);
+                       return true;
+               } else {
+                       /* Run in circles and scream like a little girl.  */
+                       while (step_.load () == step)
+                               thrd_yield();
+                       return false;
+               }
+       }
+
+ protected:
+       /* Number of synchronized threads. */
+       const unsigned int n_;
+
+       /* Number of threads currently spinning.  */
+       std::atomic<unsigned int> nwait_;
+
+       /* Number of barrier syncronizations completed so far, 
+        *      * it's OK to wrap.  */
+       std::atomic<unsigned int> step_;
+};