PaddlePaddle
diff --git a/‎paddle/fluid/pir/dialect/op_generator/op_gen.py
Copy file name to clipboard
+2Lines changed: 2 additions & 0 deletions b/‎paddle/fluid/pir/dialect/op_generator/op_gen.py
Copy file name to clipboard
+2Lines changed: 2 additions & 0 deletions
diff --git a/‎paddle/fluid/pir/dialect/op_generator/op_interface_gen.py
Copy file name to clipboard
+2-2Lines changed: 2 additions & 2 deletions b/‎paddle/fluid/pir/dialect/op_generator/op_interface_gen.py
Copy file name to clipboard
+2-2Lines changed: 2 additions & 2 deletions
diff --git a/‎paddle/fluid/pir/dialect/operator/interface/vjp.cc
Copy file name to clipboard
+1-19Lines changed: 1 addition & 19 deletions b/‎paddle/fluid/pir/dialect/operator/interface/vjp.cc
Copy file name to clipboard
+1-19Lines changed: 1 addition & 19 deletions
diff --git a/‎paddle/fluid/pir/dialect/operator/interface/vjp.h
Copy file name to clipboard
+4-11Lines changed: 4 additions & 11 deletions b/‎paddle/fluid/pir/dialect/operator/interface/vjp.h
Copy file name to clipboard
+4-11Lines changed: 4 additions & 11 deletions
diff --git a/‎paddle/fluid/pir/dialect/operator/ir/control_flow_op.cc
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions b/‎paddle/fluid/pir/dialect/operator/ir/control_flow_op.cc
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions
diff --git a/‎paddle/fluid/pir/dialect/operator/ir/control_flow_op.h
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions b/‎paddle/fluid/pir/dialect/operator/ir/control_flow_op.h
Copy file name to clipboard
+3-3Lines changed: 3 additions & 3 deletions
@@ -238,6 +238,8 @@ class {op_name} : public pir::Op<{op_name}{interfaces}{traits}> {{
     'add_n_with_kernel',
     'split_grad',
     'expand',
+    'increment',
+    'increment_',
 }
 
 attr_types_map = {
 
@@ -88,7 +88,7 @@
     }"""
 
 OP_VJP_DEFINE_TEMPLATE = """
-std::vector<std::vector<pir::OpResult>> {op_class_name}::Vjp(pir::Operation* op, const std::vector<std::vector<pir::Value>>& inputs_, const std::vector<std::vector<pir::OpResult>>& outputs, const std::vector<std::vector<pir::Value>>& out_grads, const std::vector<std::vector<bool>>& stop_gradients){{
+std::vector<std::vector<pir::OpResult>> {op_class_name}::Vjp(pir::Operation* op, const std::vector<std::vector<pir::Value>>& inputs_, const std::vector<std::vector<pir::Value>>& outputs, const std::vector<std::vector<pir::Value>>& out_grads, const std::vector<std::vector<bool>>& stop_gradients){{
 {check_param}
     VLOG(6) << "Prepare inputs of {op_grad_name}";
 {backward_input_code}
@@ -302,5 +302,5 @@ def gen_exclusive_interface_str(op_info, op_info_items):
                 "  static void InferMeta( phi::InferMetaContext *infer_meta );"
             )
     if op_info.op_phi_name[0] not in vjp_interface_black_list:
-        exclusive_interface_str += "\n  static std::vector<std::vector<pir::OpResult>> Vjp(pir::Operation* op, const std::vector<std::vector<pir::Value>>& inputs_, const std::vector<std::vector<pir::OpResult>>& outputs, const std::vector<std::vector<pir::Value>>& out_grads, const std::vector<std::vector<bool>>& stop_gradients);"
+        exclusive_interface_str += "\n  static std::vector<std::vector<pir::OpResult>> Vjp(pir::Operation* op, const std::vector<std::vector<pir::Value>>& inputs_, const std::vector<std::vector<pir::Value>>& outputs, const std::vector<std::vector<pir::Value>>& out_grads, const std::vector<std::vector<bool>>& stop_gradients);"
     return exclusive_interface_str
@@ -14,24 +14,6 @@
 
 #include "paddle/fluid/pir/dialect/operator/interface/vjp.h"
 
-namespace paddle::dialect {
-std::vector<std::vector<pir::OpResult>> VjpInterface::Vjp(
-    pir::Operation* op,
-    const std::vector<std::vector<pir::Value>>& inputs,
-    const std::vector<std::vector<pir::OpResult>>& outputs,
-    const std::vector<std::vector<pir::OpResult>>& out_grads,
-    const std::vector<std::vector<bool>>& stop_gradients) {
-  std::vector<std::vector<pir::Value>> out_grads_value;
-  for (const auto& grad : out_grads) {
-    std::vector<pir::Value> grad_value;
-    for (auto op_result : grad) {
-      grad_value.emplace_back(op_result);
-    }
-    out_grads_value.emplace_back(std::move(grad_value));
-  }
-  return impl_->vjp_(op, inputs, outputs, out_grads_value, stop_gradients);
-}
-
-}  // namespace paddle::dialect
+namespace paddle::dialect {}  // namespace paddle::dialect
 
 IR_DEFINE_EXPLICIT_TYPE_ID(paddle::dialect::VjpInterface)
@@ -23,14 +23,14 @@ class VjpInterface : public pir::OpInterfaceBase<VjpInterface> {
     explicit Concept(std::vector<std::vector<pir::OpResult>> (*vjp)(
         pir::Operation* op,
         const std::vector<std::vector<pir::Value>>& inputs,
-        const std::vector<std::vector<pir::OpResult>>& outputs,
+        const std::vector<std::vector<pir::Value>>& outputs,
         const std::vector<std::vector<pir::Value>>& out_grads,
         const std::vector<std::vector<bool>>& stop_gradients))
         : vjp_(vjp) {}
     std::vector<std::vector<pir::OpResult>> (*vjp_)(
         pir::Operation* op,
         const std::vector<std::vector<pir::Value>>& inputs,
-        const std::vector<std::vector<pir::OpResult>>& outputs,
+        const std::vector<std::vector<pir::Value>>& outputs,
         const std::vector<std::vector<pir::Value>>& out_grads,
         const std::vector<std::vector<bool>>& stop_gradients);
   };
@@ -40,7 +40,7 @@ class VjpInterface : public pir::OpInterfaceBase<VjpInterface> {
     static std::vector<std::vector<pir::OpResult>> Vjp(
         pir::Operation* op,
         const std::vector<std::vector<pir::Value>>& inputs,
-        const std::vector<std::vector<pir::OpResult>>& outputs,
+        const std::vector<std::vector<pir::Value>>& outputs,
         const std::vector<std::vector<pir::Value>>& out_grads,
         const std::vector<std::vector<bool>>& stop_gradients) {
       return ConcreteOp::Vjp(op, inputs, outputs, out_grads, stop_gradients);
@@ -56,19 +56,12 @@ class VjpInterface : public pir::OpInterfaceBase<VjpInterface> {
   std::vector<std::vector<pir::OpResult>> Vjp(
       pir::Operation* op,
       const std::vector<std::vector<pir::Value>>& inputs,
-      const std::vector<std::vector<pir::OpResult>>& outputs,
+      const std::vector<std::vector<pir::Value>>& outputs,
       const std::vector<std::vector<pir::Value>>& out_grads,
       const std::vector<std::vector<bool>>& stop_gradients) {
     return impl_->vjp_(op, inputs, outputs, out_grads, stop_gradients);
   }
 
-  std::vector<std::vector<pir::OpResult>> Vjp(
-      pir::Operation* op,
-      const std::vector<std::vector<pir::Value>>& inputs,
-      const std::vector<std::vector<pir::OpResult>>& outputs,
-      const std::vector<std::vector<pir::OpResult>>& out_grads,
-      const std::vector<std::vector<bool>>& stop_gradients);
-
  private:
   Concept* impl_;
 };
 
@@ -246,7 +246,7 @@ void IfOp::VerifyRegion() {
 std::vector<std::vector<pir::OpResult>> IfOp::Vjp(
     pir::Operation *op,
     const std::vector<std::vector<pir::Value>> &inputs_,
-    const std::vector<std::vector<pir::OpResult>> &outputs,
+    const std::vector<std::vector<pir::Value>> &outputs,
     const std::vector<std::vector<pir::Value>> &out_grads,
     const std::vector<std::vector<bool>> &stop_gradients) {
   PADDLE_ENFORCE_EQ(
@@ -346,7 +346,7 @@ void WhileOp::Print(pir::IrPrinter &printer) {
 std::vector<std::vector<pir::OpResult>> WhileOp::Vjp(
     pir::Operation *op,
     const std::vector<std::vector<pir::Value>> &inputs,
-    const std::vector<std::vector<pir::OpResult>> &outputs,
+    const std::vector<std::vector<pir::Value>> &outputs,
     const std::vector<std::vector<pir::Value>> &out_grads,
     const std::vector<std::vector<bool>> &stop_gradients) {
   auto fwd_op = WhileOp::dyn_cast(op);
@@ -417,7 +417,7 @@ std::vector<std::vector<pir::OpResult>> WhileOp::Vjp(
 std::vector<std::vector<pir::OpResult>> TuplePushOpVjpInterfaceModel::Vjp(
     pir::Operation *op,
     const std::vector<std::vector<pir::Value>> &inputs,
-    const std::vector<std::vector<pir::OpResult>> &outputs,
+    const std::vector<std::vector<pir::Value>> &outputs,
     const std::vector<std::vector<pir::Value>> &out_grads,
     const std::vector<std::vector<bool>> &stop_gradients) {
   PADDLE_ENFORCE_EQ(
 
@@ -52,7 +52,7 @@ class IfOp : public pir::Op<IfOp, VjpInterface> {
   static std::vector<std::vector<pir::OpResult>> Vjp(
       pir::Operation *op,
       const std::vector<std::vector<pir::Value>> &inputs_,
-      const std::vector<std::vector<pir::OpResult>> &outputs,
+      const std::vector<std::vector<pir::Value>> &outputs,
       const std::vector<std::vector<pir::Value>> &out_grads,
       const std::vector<std::vector<bool>> &stop_gradients);
 };
@@ -87,7 +87,7 @@ class WhileOp : public pir::Op<WhileOp, VjpInterface> {
   static std::vector<std::vector<pir::OpResult>> Vjp(
       pir::Operation *op,
       const std::vector<std::vector<pir::Value>> &inputs_,
-      const std::vector<std::vector<pir::OpResult>> &outputs,
+      const std::vector<std::vector<pir::Value>> &outputs,
       const std::vector<std::vector<pir::Value>> &out_grads,
       const std::vector<std::vector<bool>> &stop_gradients);
 };
@@ -96,7 +96,7 @@ struct TuplePushOpVjpInterfaceModel : public VjpInterface::Concept {
   static std::vector<std::vector<pir::OpResult>> Vjp(
       pir::Operation *op,
       const std::vector<std::vector<pir::Value>> &inputs,
-      const std::vector<std::vector<pir::OpResult>> &outputs,
+      const std::vector<std::vector<pir::Value>> &outputs,
       const std::vector<std::vector<pir::Value>> &out_grads,
       const std::vector<std::vector<bool>> &stop_gradients);
Original file line number	Diff line number	Diff line change
`@@ -238,6 +238,8 @@ class {op_name} : public pir::Op<{op_name}{interfaces}{traits}> {{`
`238`	`238`	`'add_n_with_kernel',`
`239`	`239`	`'split_grad',`
`240`	`240`	`'expand',`
	`241`	`+ 'increment',`
	`242`	`+ 'increment_',`
`241`	`243`	`}`
`242`	`244`
`243`	`245`	`attr_types_map = {`