fix atomic issue when cpu only

7 years ago · e0b17754db
parent 93db58bbfa
commit e0b17754db
1 changed files with 1 additions and 0 deletions
--- a/paddle/framework/operator.cc
+++ b/paddle/framework/operator.cc
@ -14,6 +14,7 @@ limitations under the License. */

 #include "paddle/framework/operator.h"
 #include <algorithm>
+#include <atomic>

 namespace paddle {
 namespace framework {