From 729b2152ae7ee525767ea780473208ce2f627ba5 Mon Sep 17 00:00:00 2001
From: qiaolongfei <qiaolongfei@baidu.com>
Date: Tue, 26 Sep 2017 23:44:57 -0700
Subject: [PATCH] add include op_registry.h to elementwise_op_function.h

---
 paddle/operators/elementwise_op.h          | 1 +
 paddle/operators/elementwise_op_function.h | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/paddle/operators/elementwise_op.h b/paddle/operators/elementwise_op.h
index d7013654fd..3082f37422 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 86f5fa376c..3eb97f60b5 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 {
-- 
GitLab