diff --git a/paddle/fluid/framework/operator.h b/paddle/fluid/framework/operator.h
index e34a0e21415c3615d0ce9b85f8a17a230347949d..3c3e9096c0da575b1719eb142676627fb0150754 100644
--- a/paddle/fluid/framework/operator.h
+++ b/paddle/fluid/framework/operator.h
@@ -16,9 +16,11 @@ limitations under the License. */
 
 #include <algorithm>
 #include <atomic>
+#include <memory>
 #include <string>
 #include <tuple>
 #include <unordered_map>
+#include <utility>
 #include <vector>
 
 #include "glog/logging.h"  // For VLOG