diff --git a/paddle/operators/elementwise_op.h b/paddle/operators/elementwise_op.h
index d7013654fd98bab56070874d30ed150d9a0c8a4d..3082f37422faa990bbf03c8a1a87b025d481a290 100644
--- a/paddle/operators/elementwise_op.h
+++ b/paddle/operators/elementwise_op.h
@@ -14,6 +14,7 @@ limitations under the License. */
 
 #pragma once
 #include "paddle/framework/op_registry.h"
+#include "paddle/framework/operator.h"
 
 namespace paddle {
 namespace operators {
diff --git a/paddle/operators/elementwise_op_function.h b/paddle/operators/elementwise_op_function.h
index 86f5fa376cecbf6b89cf4ed5d650c6e7ccf011a1..3eb97f60b59848d23bcd15ea1e3d2f21b721f6a4 100644
--- a/paddle/operators/elementwise_op_function.h
+++ b/paddle/operators/elementwise_op_function.h
@@ -14,7 +14,9 @@
 
 #pragma once
 #include "paddle/framework/eigen.h"
+#include "paddle/framework/op_registry.h"
 #include "paddle/framework/operator.h"
+
 #include "paddle/operators/math/math_function.h"
 
 namespace paddle {