fix test_decayed_adagrad_op

paddle/fluid/pir/dialect/op_generator/ops_api_gen.py

@@ -110,6 +110,7 @@
     'c_identity',
     'c_reduce_sum',
     'c_reducescatter',
+    'decayed_adagrad',
     'dpsgd',
     'embedding_grad_sparse',
     'fused_batch_norm_act_',

paddle/fluid/pir/dialect/operator/ir/ops.yaml

@@ -309,6 +309,15 @@
   output : Tensor(out)
   invoke : copy_to_impl(x, place, blocking)
 
+- op : decayed_adagrad
+  args : (Tensor param, Tensor grad, Tensor moment, Tensor learning_rate, float decay = 0.95f, float epsilon = 1.0e-6f)
+  output : Tensor(param_out), Tensor(moment_out)
+  infer_meta :
+    func : DecayedAdagradInferMeta
+  kernel :
+    func : decayed_adagrad
+    data_type : param
+
 - op : decode_jpeg
   args : (Tensor x, str mode, Place place)
   output : Tensor(out)

paddle/phi/api/yaml/op_compat.yaml

@@ -3345,6 +3345,12 @@
   outputs:
     {out: Out}
 
+- op: decayed_adagrad
+  inputs:
+    {param : Param, grad : Grad, moment : Moment, learning_rate : LearningRate}
+  outputs:
+    {param_out : ParamOut, moment_out : MomentOut}
+
 - op: dpsgd
   inputs:
     {param: Param,grad: Grad,learning_rate: LearningRate}

test/white_list/pir_op_test_white_list

@@ -77,6 +77,7 @@ test_cummax_op
 test_cummin_op
 test_cumprod_op
 test_cumsum_op
+test_decayed_adagrad_op
 test_deformable_conv_op
 test_dequantize_mkldnn_op
 test_determinant_op