From e5600e8e2029404e131286d9c30be96a3fb0e96e Mon Sep 17 00:00:00 2001
From: barrierye <barriery@qq.com>
Date: Wed, 21 Nov 2018 10:50:46 +0800
Subject: [PATCH] fix bug that finish_set_filelist failed in multithread

---
 paddle/fluid/framework/data_feed.cc | 1 +
 paddle/fluid/framework/data_feed.h  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/framework/data_feed.cc b/paddle/fluid/framework/data_feed.cc
index 556591acdd2..a1b718fe2c4 100755
--- a/paddle/fluid/framework/data_feed.cc
+++ b/paddle/fluid/framework/data_feed.cc
@@ -41,6 +41,7 @@ namespace framework {
 std::vector<std::string> DataFeed::filelist_;
 size_t DataFeed::file_idx_;
 std::mutex DataFeed::mutex_for_pick_file_;
+bool DataFeed::finish_set_filelist_;
 
 void DataFeed::AddFeedVar(Variable* var, const std::string& name) {
   CheckInit();
diff --git a/paddle/fluid/framework/data_feed.h b/paddle/fluid/framework/data_feed.h
index 41cc9659273..f37d22b7dc6 100755
--- a/paddle/fluid/framework/data_feed.h
+++ b/paddle/fluid/framework/data_feed.h
@@ -205,7 +205,7 @@ class DataFeed {
   int batch_size_;
 
   bool finish_init_;
-  bool finish_set_filelist_;
+  static bool finish_set_filelist_;
   bool finish_binding_memory_;
   bool finish_start_;
 };
-- 
GitLab