【PIR OpTest Fix No.27】 fix test_softmax_mask_fuse_op (#62767)

Eddie-Wang1120 · web-flow · commit a29a7546c00f · 2024-03-19T16:10:21.000+08:00
* fix fused_softmax_mask_op

* add to whitelist

* update fix
diff --git a/paddle/fluid/ir_adaptor/translator/op_compat_gen.py b/paddle/fluid/ir_adaptor/translator/op_compat_gen.py
@@ -164,6 +164,7 @@ def insert_new_mutable_attributes(
         "atol_tensor": "TolTensor",
         "out": "Out",
     }
+    op_arg_name_mappings['fused_softmax_mask_grad'].update({"out": "Softmax"})
     op_arg_name_mappings['push_sparse_v2'].update(
         {"out_grad_in": "Out@GRAD", "out_grad_out": "Out@GRAD"}
     )
diff --git a/paddle/phi/api/yaml/op_compat.yaml b/paddle/phi/api/yaml/op_compat.yaml
@@ -3706,6 +3706,13 @@
   attrs :
     {scale_x : Scale_x, scale_y : Scale_y, scale_out : Scale_out, scale_in_eltwise : Scale_in_eltwise, fused_reshape_x : fused_reshape_X, fused_transpose_x : fused_transpose_X, fused_reshape_y : fused_reshape_Y, fused_transpose_y : fused_transpose_Y, fused_reshape_out : fused_reshape_Out, fused_transpose_out : fused_transpose_Out}
 
+- op: fused_softmax_mask
+  backward : fused_softmax_mask_grad
+  inputs :
+    {x: X, mask: Mask}
+  outputs :
+    {out : Out}
+
 - op: fused_softplus
   inputs :
     {x: X}
diff --git a/test/white_list/pir_op_test_white_list b/test/white_list/pir_op_test_white_list
@@ -245,6 +245,7 @@ test_sigmoid_cross_entropy_with_logits_op
 test_sign_op
 test_size_op
 test_slice_op
+test_softmax_mask_fuse_op
 test_softmax_mask_fuse_upper_triangle_op
 test_softmax_op
 test_solve_op

Original file line number	Diff line number	Diff line change
`@@ -164,6 +164,7 @@ def insert_new_mutable_attributes(`
`164`	`164`	`"atol_tensor": "TolTensor",`
`165`	`165`	`"out": "Out",`
`166`	`166`	`}`
	`167`	`+ op_arg_name_mappings['fused_softmax_mask_grad'].update({"out": "Softmax"})`
`167`	`168`	`op_arg_name_mappings['push_sparse_v2'].update(`
`168`	`169`	`{"out_grad_in": "Out@GRAD", "out_grad_out": "Out@GRAD"}`
`169`	`170`	`)`