git.sesse.net Git - stockfish/blob - src/thread_win32_osx.h

   1 /*
   2   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
   3   Copyright (C) 2004-2023 The Stockfish developers (see AUTHORS file)
   4
   5   Stockfish is free software: you can redistribute it and/or modify
   6   it under the terms of the GNU General Public License as published by
   7   the Free Software Foundation, either version 3 of the License, or
   8   (at your option) any later version.
   9
  10   Stockfish is distributed in the hope that it will be useful,
  11   but WITHOUT ANY WARRANTY; without even the implied warranty of
  12   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13   GNU General Public License for more details.
  14
  15   You should have received a copy of the GNU General Public License
  16   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #ifndef THREAD_WIN32_OSX_H_INCLUDED
  20 #define THREAD_WIN32_OSX_H_INCLUDED
  21
  22 #include <thread>
  23
  24 // On OSX threads other than the main thread are created with a reduced stack
  25 // size of 512KB by default, this is too low for deep searches, which require
  26 // somewhat more than 1MB stack, so adjust it to TH_STACK_SIZE.
  27 // The implementation calls pthread_create() with the stack size parameter
  28 // equal to the Linux 8MB default, on platforms that support it.
  29
  30 #if defined(__APPLE__) || defined(__MINGW32__) || defined(__MINGW64__) || defined(USE_PTHREADS)
  31
  32     #include <pthread.h>
  33
  34 namespace Stockfish {
  35
  36 static const size_t TH_STACK_SIZE = 8 * 1024 * 1024;
  37
  38 template<class T, class P = std::pair<T*, void (T::*)()>>
  39 void* start_routine(void* ptr) {
  40     P* p = reinterpret_cast<P*>(ptr);
  41     (p->first->*(p->second))();  // Call member function pointer
  42     delete p;
  43     return nullptr;
  44 }
  45
  46 class NativeThread {
  47
  48     pthread_t thread;
  49
  50    public:
  51     template<class T, class P = std::pair<T*, void (T::*)()>>
  52     explicit NativeThread(void (T::*fun)(), T* obj) {
  53         pthread_attr_t attr_storage, *attr = &attr_storage;
  54         pthread_attr_init(attr);
  55         pthread_attr_setstacksize(attr, TH_STACK_SIZE);
  56         pthread_create(&thread, attr, start_routine<T>, new P(obj, fun));
  57     }
  58     void join() { pthread_join(thread, nullptr); }
  59 };
  60
  61 }  // namespace Stockfish
  62
  63 #else  // Default case: use STL classes
  64
  65 namespace Stockfish {
  66
  67 using NativeThread = std::thread;
  68
  69 }  // namespace Stockfish
  70
  71 #endif
  72
  73 #endif  // #ifndef THREAD_WIN32_OSX_H_INCLUDED