【PIR】add batch_norm_grad_grad in pir (#59373)

yangguohao · web-flow · commit a4a66c8026e4 · 2023-11-29T15:18:41.000+08:00
* add batch_norm_grad_grad in pir

* fix

* fix 2023-11-27
diff --git a/paddle/fluid/ir_adaptor/translator/op_compat_gen.py b/paddle/fluid/ir_adaptor/translator/op_compat_gen.py
@@ -136,6 +136,17 @@ def insert_new_mutable_attributes(
         "grad_y_grad": "DDY",
         "grad_out_grad": "DDOut",
     }
+    op_arg_name_mappings["batch_norm_grad_grad"] = {
+        "scale_grad": "DScale",
+        "x_grad": "DX",
+        "grad_out_grad": "DDY",
+        "out_mean": "OutMean",
+        "out_variance": "OutVariance",
+        "grad_x_grad": "DDX",
+        "grad_scale_grad": "DDScale",
+        "grad_bias_grad": "DDBias",
+        "grad_out": "DY",
+    }
 
     op_name_normailzer_template = env.get_template("op_compat_info.cc.j2")
     with open(output_source_file, 'wt') as f:
diff --git a/paddle/phi/api/yaml/op_compat.yaml b/paddle/phi/api/yaml/op_compat.yaml
@@ -318,7 +318,7 @@
     {auc : AUC, stat_pos_out : StatPosOut, stat_neg_out : StatNegOut}
 
 - op : batch_norm
-  backward : batch_norm_grad
+  backward : batch_norm_grad, batch_norm_double_grad(batch_norm_grad_grad)
   inputs:
     x : X
     mean : Mean
