diff --git a/paddle/phi/core/utils/data_type.h b/paddle/phi/core/utils/data_type.h
index a7bf1123a05a03070d87c93689dda2e8fd399ded..7852d87c9a293bfe07bfdae3ef34be19cad5ca47 100644
--- a/paddle/phi/core/utils/data_type.h
+++ b/paddle/phi/core/utils/data_type.h
@@ -20,8 +20,6 @@ limitations under the License. */
 
 #include "paddle/phi/common/data_type.h"
 #include "paddle/phi/core/enforce.h"
-#include "paddle/phi/kernels/funcs/eigen/extensions.h"
-
 namespace phi {
 
 // Here we can't depend on the fluid proto::VarType, so we use the dtype enum
diff --git a/paddle/phi/kernels/funcs/eigen/eigen_function.h b/paddle/phi/kernels/funcs/eigen/eigen_function.h
index 1e81256e79e143d21e2681d9f91beec5e73741aa..42f46b814e1682c1616505c05bc9d20aa7a37f24 100644
--- a/paddle/phi/kernels/funcs/eigen/eigen_function.h
+++ b/paddle/phi/kernels/funcs/eigen/eigen_function.h
@@ -18,6 +18,8 @@ limitations under the License. */
 #ifndef NOMINMAX
 #define NOMINMAX
 #endif
+
+#include "paddle/phi/kernels/funcs/eigen/extensions.h"
 #include "unsupported/Eigen/CXX11/Tensor"
 
 namespace phi {