diff --git a/paddle/phi/api/yaml/generator/templates/op.c.j2 b/paddle/phi/api/yaml/generator/templates/op.c.j2
index 7f13eb958258987f3110c401b500632d1ee37f29..0c2708ce223c7c2bef271cafa75150d8dae0084a 100644
--- a/paddle/phi/api/yaml/generator/templates/op.c.j2
+++ b/paddle/phi/api/yaml/generator/templates/op.c.j2
@@ -11,7 +11,7 @@
 #include "paddle/phi/infermeta/binary.h"
 #include "paddle/phi/infermeta/ternary.h"
 #include "paddle/phi/infermeta/multiary.h"
-#include "paddle/phi/infermeta/backward.cc"
+#include "paddle/phi/infermeta/backward.h"
 
 namespace paddle {
 namespace operators {