add include op_registry.h to elementwise_op_function.h

8 years ago · 729b2152ae
parent abdcd8282e
commit 729b2152ae
2 changed files with 3 additions and 0 deletions
--- 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 {
--- 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 {