Clear Workers after changing the network

official-stockfish · May 1, 2024 · 1e8aac5 · 1e8aac5
1 parent be14233
commit 1e8aac5
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/engine.cpp b/src/engine.cpp
@@ -141,14 +141,19 @@ void Engine::verify_networks() const {
 }
 
 void Engine::load_networks() {
-    networks.big.load(binaryDirectory, options["EvalFile"]);
-    networks.small.load(binaryDirectory, options["EvalFileSmall"]);
+    load_big_network(options["EvalFile"]);
+    load_small_network(options["EvalFileSmall"]);
+    threads.clear();
 }
 
-void Engine::load_big_network(const std::string& file) { networks.big.load(binaryDirectory, file); }
+void Engine::load_big_network(const std::string& file) {
+    networks.big.load(binaryDirectory, file);
+    threads.clear();
+}
 
 void Engine::load_small_network(const std::string& file) {
     networks.small.load(binaryDirectory, file);
+    threads.clear();
 }
 
 void Engine::save_network(const std::pair<std::optional<std::string>, std::string> files[2]) {

diff --git a/src/thread.cpp b/src/thread.cpp
@@ -161,6 +161,9 @@ void ThreadPool::clear() {
     for (Thread* th : threads)
         th->worker->clear();
 
+    if (threads.size() == 0)
+        return;
+
     main_manager()->callsCnt                 = 0;
     main_manager()->bestPreviousScore        = VALUE_INFINITE;
     main_manager()->bestPreviousAverageScore = VALUE_INFINITE;