1 files changed, 93 insertions, 0 deletions
diff --git a/boost/fiber/algo/numa/work_stealing.hpp b/boost/fiber/algo/numa/work_stealing.hpp
new file mode 100644
index 0000000000..26032ab35e
--- /dev/null
+++ b/boost/fiber/algo/numa/work_stealing.hpp
@@ -0,0 +1,93 @@
+
+//          Copyright Oliver Kowalke 2017.
+// Distributed under the Boost Software License, Version 1.0.
+//    (See accompanying file LICENSE_1_0.txt or copy at
+//          http://www.boost.org/LICENSE_1_0.txt)
+//
+
+#ifndef BOOST_FIBERS_ALGO_NUMA_WORK_STEALING_H
+#define BOOST_FIBERS_ALGO_NUMA_WORK_STEALING_H
+
+#include <condition_variable>
+#include <chrono>
+#include <cstddef>
+#include <cstdint>
+#include <mutex>
+#include <vector>
+
+#include <boost/config.hpp>
+#include <boost/intrusive_ptr.hpp>
+
+#include <boost/fiber/algo/algorithm.hpp>
+#include <boost/fiber/context.hpp>
+#include <boost/fiber/detail/config.hpp>
+#include <boost/fiber/detail/context_spinlock_queue.hpp>
+#include <boost/fiber/detail/context_spmc_queue.hpp>
+#include <boost/fiber/numa/pin_thread.hpp>
+#include <boost/fiber/numa/topology.hpp>
+#include <boost/fiber/scheduler.hpp>
+
+#ifdef BOOST_HAS_ABI_HEADERS
+#  include BOOST_ABI_PREFIX
+#endif
+
+namespace boost {
+namespace fibers {
+namespace algo {
+namespace numa {
+
+class work_stealing : public algorithm {
+private:
+    static std::vector< intrusive_ptr< work_stealing > >    schedulers_;
+
+    std::uint32_t                                           cpu_id_;
+    std::vector< std::uint32_t >                            local_cpus_;
+    std::vector< std::uint32_t >                            remote_cpus_;
+#ifdef BOOST_FIBERS_USE_SPMC_QUEUE
+    detail::context_spmc_queue                              rqueue_{};
+#else
+    detail::context_spinlock_queue                          rqueue_{};
+#endif
+    std::mutex                                              mtx_{};
+    std::condition_variable                                 cnd_{};
+    bool                                                    flag_{ false };
+    bool                                                    suspend_;
+
+    static void init_( std::vector< boost::fibers::numa::node > const&,
+                       std::vector< intrusive_ptr< work_stealing > > &);
+
+public:
+    work_stealing( std::uint32_t, std::uint32_t,
+                   std::vector< boost::fibers::numa::node > const&,
+                   bool = false);
+
+    work_stealing( work_stealing const&) = delete;
+    work_stealing( work_stealing &&) = delete;
+
+    work_stealing & operator=( work_stealing const&) = delete;
+    work_stealing & operator=( work_stealing &&) = delete;
+
+    virtual void awakened( context *) noexcept;
+
+    virtual context * pick_next() noexcept;
+
+    virtual context * steal() noexcept {
+        return rqueue_.steal();
+    }
+
+    virtual bool has_ready_fibers() const noexcept {
+        return ! rqueue_.empty();
+    }
+
+    virtual void suspend_until( std::chrono::steady_clock::time_point const&) noexcept;
+
+    virtual void notify() noexcept;
+};
+
+}}}}
+
+#ifdef BOOST_HAS_ABI_HEADERS
+#  include BOOST_ABI_SUFFIX
+#endif
+
+#endif // BOOST_FIBERS_ALGO_NUMA_WORK_STEALING_H