mirror of
https://github.com/sockspls/badfish
synced 2025-04-30 16:53:09 +00:00
7ffae17f85
fixes #3350 and is a small cleanup that might make it easier to use SF in separate projects, like a NNUE trainer or similar. closes https://github.com/official-stockfish/Stockfish/pull/3370 No functional change.
74 lines
2.2 KiB
C++
74 lines
2.2 KiB
C++
/*
|
|
Stockfish, a UCI chess playing engine derived from Glaurung 2.1
|
|
Copyright (C) 2004-2021 The Stockfish developers (see AUTHORS file)
|
|
|
|
Stockfish is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
Stockfish is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef THREAD_WIN32_OSX_H_INCLUDED
|
|
#define THREAD_WIN32_OSX_H_INCLUDED
|
|
|
|
#include <thread>
|
|
|
|
/// On OSX threads other than the main thread are created with a reduced stack
|
|
/// size of 512KB by default, this is too low for deep searches, which require
|
|
/// somewhat more than 1MB stack, so adjust it to TH_STACK_SIZE.
|
|
/// The implementation calls pthread_create() with the stack size parameter
|
|
/// equal to the linux 8MB default, on platforms that support it.
|
|
|
|
#if defined(__APPLE__) || defined(__MINGW32__) || defined(__MINGW64__) || defined(USE_PTHREADS)
|
|
|
|
#include <pthread.h>
|
|
|
|
namespace Stockfish {
|
|
|
|
static const size_t TH_STACK_SIZE = 8 * 1024 * 1024;
|
|
|
|
template <class T, class P = std::pair<T*, void(T::*)()>>
|
|
void* start_routine(void* ptr)
|
|
{
|
|
P* p = reinterpret_cast<P*>(ptr);
|
|
(p->first->*(p->second))(); // Call member function pointer
|
|
delete p;
|
|
return NULL;
|
|
}
|
|
|
|
class NativeThread {
|
|
|
|
pthread_t thread;
|
|
|
|
public:
|
|
template<class T, class P = std::pair<T*, void(T::*)()>>
|
|
explicit NativeThread(void(T::*fun)(), T* obj) {
|
|
pthread_attr_t attr_storage, *attr = &attr_storage;
|
|
pthread_attr_init(attr);
|
|
pthread_attr_setstacksize(attr, TH_STACK_SIZE);
|
|
pthread_create(&thread, attr, start_routine<T>, new P(obj, fun));
|
|
}
|
|
void join() { pthread_join(thread, NULL); }
|
|
};
|
|
|
|
} // namespace Stockfish
|
|
|
|
#else // Default case: use STL classes
|
|
|
|
namespace Stockfish {
|
|
|
|
typedef std::thread NativeThread;
|
|
|
|
} // namespace Stockfish
|
|
|
|
#endif
|
|
|
|
#endif // #ifndef THREAD_WIN32_OSX_H_INCLUDED
|