diff --git a/src/thread.h b/src/thread.h index f8e65941..ea645b45 100644 --- a/src/thread.h +++ b/src/thread.h @@ -39,16 +39,20 @@ const size_t MAX_THREADS = 128; const size_t MAX_SPLITPOINTS_PER_THREAD = 8; const size_t MAX_SLAVES_PER_SPLITPOINT = 4; -/// Spinlock class wraps low level atomic operations to provide spin lock functionality + +/// Spinlock class wraps low level atomic operations to provide a spin lock class Spinlock { - std::atomic_flag lock; + std::atomic_int lock; public: - Spinlock() { std::atomic_flag_clear(&lock); } - void acquire() { while (lock.test_and_set(std::memory_order_acquire)) {} } - void release() { lock.clear(std::memory_order_release); } + Spinlock() { lock = 1; } // Init here to workaround a bug with MSVC 2013 + void acquire() { + while (lock.fetch_sub(1, std::memory_order_acquire) != 1) + while (lock.load(std::memory_order_relaxed) <= 0) {} + } + void release() { lock.store(1, std::memory_order_release); } };