From e2130502234f042a6381939f80c640bbebe2e1c6 Mon Sep 17 00:00:00 2001
From: Wang Guibao <wang_guibao@163.com>
Date: Thu, 13 Dec 2018 20:49:50 +0800
Subject: [PATCH] Fix multi-threading bug with WItH_MKL=ON (#14882)

fixes #14884
---
 paddle/fluid/framework/executor_thread_worker.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/fluid/framework/executor_thread_worker.cc b/paddle/fluid/framework/executor_thread_worker.cc
index 3d53511615..5fc5aeb662 100644
--- a/paddle/fluid/framework/executor_thread_worker.cc
+++ b/paddle/fluid/framework/executor_thread_worker.cc
@@ -26,6 +26,7 @@ limitations under the License. */
 #include "paddle/fluid/framework/reader.h"
 #include "paddle/fluid/framework/variable_helper.h"
 #include "paddle/fluid/inference/io.h"
+#include "paddle/fluid/platform/cpu_helper.h"
 #include "paddle/fluid/platform/place.h"
 #include "paddle/fluid/pybind/pybind.h"
 namespace paddle {
@@ -174,6 +175,8 @@ void print_fetch_var(Scope* scope, std::string var_name) {
 }
 
 void ExecutorThreadWorker::TrainFiles() {
+  platform::SetNumThreads(1);
+
   // todo: configurable
   SetDevice();
 
-- 
GitLab