@@ -18,6 +18,7 @@
 #include <string>
 
 #include "paddle/fluid/pir/drr/api/tensor_interface.h"
+#include "paddle/fluid/pir/drr/ir_operation.h"
 
 namespace paddle {
 namespace drr {