use atomic stop condition

fastmachinelearning · Sep 18, 2018 · 638b206 · 638b206
1 parent a3fb784
commit 638b206
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/AnalysisFW/interface/TFClientRemote.h b/AnalysisFW/interface/TFClientRemote.h
@@ -9,6 +9,7 @@
 #include <condition_variable>
 #include <mutex>
 #include <thread>
+#include <atomic>
 
 #include "grpc++/create_channel.h"
 #include "tensorflow_serving/apis/prediction_service.grpc.pb.h"
@@ -35,7 +36,7 @@ class JetImageData {
 	private:
 		void waitForNext();
 
-		bool stop_;
+		std::atomic<bool> stop_;
 		std::unique_ptr<std::thread> thread_;
 };