diff --git a/paddle/fluid/operators/generator/CMakeLists.txt b/paddle/fluid/operators/generator/CMakeLists.txt
index 39d90aab5af197403aea1c9d5e4f4a91ebf2b35c..53cbb990dc1ee99d0194e99506fc690e954371dc 100644
--- a/paddle/fluid/operators/generator/CMakeLists.txt
+++ b/paddle/fluid/operators/generator/CMakeLists.txt
@@ -207,7 +207,7 @@ endif()
 
 # op extra info file
 set(ops_extra_info_gen_file
-    ${CMAKE_SOURCE_DIR}/paddle/phi/api/yaml/generator/ops_extra_info_gen.py)
+    ${CMAKE_SOURCE_DIR}/paddle/fluid/operators/generator/ops_extra_info_gen.py)
 set(op_compat_yaml_file ${CMAKE_SOURCE_DIR}/paddle/phi/api/yaml/op_compat.yaml)
 set(ops_extra_info_file
     ${CMAKE_SOURCE_DIR}/paddle/fluid/operators/ops_extra_info.cc)
diff --git a/paddle/phi/api/yaml/generator/ops_extra_info_gen.py b/paddle/fluid/operators/generator/ops_extra_info_gen.py
similarity index 98%
rename from paddle/phi/api/yaml/generator/ops_extra_info_gen.py
rename to paddle/fluid/operators/generator/ops_extra_info_gen.py
index 9109399f9276ba4247eeeedd63dd07e591ab9669..11856daeba2fa4c19707d3b1b576ce131a392ea4 100644
--- a/paddle/phi/api/yaml/generator/ops_extra_info_gen.py
+++ b/paddle/fluid/operators/generator/ops_extra_info_gen.py
@@ -19,7 +19,7 @@ import yaml
 
 
 def map_code_template(attrs_str, attrs_checker_str):
-    return f"""// This file is generated by paddle/phi/api/yaml/generator/ops_extra_info_gen.py
+    return f"""// This file is generated by paddle/fluid/operators/generator/ops_extra_info_gen.py
 #include "paddle/fluid/operators/ops_extra_info.h"
 
 #include "paddle/phi/backends/gpu/cuda/cudnn_workspace_helper.h"