X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fthread.h;h=abb7a22333800f4aa2d648290640f71261e749ab;hp=8f930149f333e764d3525fc9d93fd8f4dba3798b;hb=00d9e9fd283b31e63389af091b158dbc3fedfc0e;hpb=d71f7070406bc018e3ca868461d77a90eaebfcbc diff --git a/src/thread.h b/src/thread.h index 8f930149..abb7a223 100644 --- a/src/thread.h +++ b/src/thread.h @@ -37,68 +37,23 @@ struct Thread; const size_t MAX_THREADS = 128; -const size_t MAX_SPLITPOINTS_PER_THREAD = 8; -const size_t MAX_SLAVES_PER_SPLITPOINT = 4; - -class Spinlock { - std::atomic_int _lock; - -public: - Spinlock() { _lock = 1; } // Init here to workaround a bug with MSVC 2013 - void lock() { - while (_lock.fetch_sub(1, std::memory_order_acquire) != 1) - for (int cnt = 0; _lock.load(std::memory_order_relaxed) <= 0; ++cnt) - if (cnt >= 10000) std::this_thread::yield(); // Be nice to hyperthreading - } - void unlock() { _lock.store(1, std::memory_order_release); } -}; - - -/// SplitPoint struct stores information shared by the threads searching in -/// parallel below the same split point. It is populated at splitting time. - -struct SplitPoint { - - // Const data after split point has been setup - const Position* pos; - Search::Stack* ss; - Thread* master; - Depth depth; - Value beta; - int nodeType; - bool cutNode; - - // Const pointers to shared data - MovePicker* movePicker; - SplitPoint* parentSplitPoint; - - // Shared variable data - Spinlock mutex; - std::bitset slavesMask; - volatile bool allSlavesSearching; - volatile uint64_t nodes; - volatile Value alpha; - volatile Value bestValue; - volatile Move bestMove; - volatile int moveCount; - volatile bool cutoff; -}; /// ThreadBase struct is the base of the hierarchy from where we derive all the /// specialized thread classes. -struct ThreadBase { +struct ThreadBase : public std::thread { + ThreadBase() { exit = false; } virtual ~ThreadBase() = default; virtual void idle_loop() = 0; void notify_one(); - void wait_for(volatile const bool& b); + void wait(std::atomic& b); + void wait_while(std::atomic& b); - std::thread nativeThread; Mutex mutex; ConditionVariable sleepCondition; - volatile bool exit = false; + std::atomic exit; }; @@ -111,23 +66,21 @@ struct Thread : public ThreadBase { Thread(); virtual void idle_loop(); - bool cutoff_occurred() const; - bool can_join(const SplitPoint* sp) const; + void search(bool isMainThread = false); - void split(Position& pos, Search::Stack* ss, Value alpha, Value beta, Value* bestValue, Move* bestMove, - Depth depth, int moveCount, MovePicker* movePicker, int nodeType, bool cutNode); - - SplitPoint splitPoints[MAX_SPLITPOINTS_PER_THREAD]; Pawns::Table pawnsTable; Material::Table materialTable; Endgames endgames; - Position* activePosition; - size_t idx; + size_t idx, PVIdx; int maxPly; - SplitPoint* volatile activeSplitPoint; - volatile size_t splitPointsSize; - volatile bool searching; - Spinlock allocMutex; + std::atomic searching; + + Position rootPos; + Search::RootMoveVector rootMoves; + Depth rootDepth; + Search::Stack stack[MAX_PLY+4]; + HistoryStats history; + MovesStats counterMoves; }; @@ -135,8 +88,11 @@ struct Thread : public ThreadBase { /// special threads: the main one and the recurring timer. struct MainThread : public Thread { + MainThread() { thinking = true; } // Avoid a race with start_thinking() virtual void idle_loop(); - volatile bool thinking = true; // Avoid a race with start_thinking() + void join(); + void think(); + std::atomic thinking; }; struct TimerThread : public ThreadBase { @@ -160,12 +116,8 @@ struct ThreadPool : public std::vector { MainThread* main() { return static_cast(at(0)); } void read_uci_options(); - Thread* available_slave(const SplitPoint* sp) const; - void wait_for_think_finished(); void start_thinking(const Position&, const Search::LimitsType&, Search::StateStackPtr&); - - Depth minimumSplitDepth; - ConditionVariable sleepCondition; + int64_t nodes_searched(); TimerThread* timer; };