diff --git a/paddle/phi/api/yaml/backward.yaml b/paddle/phi/api/yaml/backward.yaml
index 2ea1660e746c9f9f93f3a24f81f0fe72f07f9c04..a109fac21581571b10c8ff8ffa058aac36047508 100644
--- a/paddle/phi/api/yaml/backward.yaml
+++ b/paddle/phi/api/yaml/backward.yaml
@@ -1700,9 +1700,9 @@
   optional : boxes_num
 
 - backward_op : put_along_axis_grad
-  forward : put_along_axis (Tensor arr, Tensor indices, Tensor value, int axis, str reduce = "assign") -> Tensor(out)
+  forward : put_along_axis (Tensor arr, Tensor indices, Tensor values, int axis, str reduce = "assign") -> Tensor(out)
   args : (Tensor arr, Tensor indices, Tensor out_grad, int axis, str reduce)
-  output : Tensor(arr_grad), Tensor(value_grad)
+  output : Tensor(arr_grad), Tensor(values_grad)
   infer_meta :
     func : GeneralBinaryGradInferMeta
     param : [arr, indices]
diff --git a/test/white_list/new_ir_op_test_white_list b/test/white_list/new_ir_op_test_white_list
index 40cfd7c2dfb733563704717a15fa8b4d5a7b2477..b17b87a0a5865114b2a77d573be14f7b27cb7108 100644
--- a/test/white_list/new_ir_op_test_white_list
+++ b/test/white_list/new_ir_op_test_white_list
@@ -137,6 +137,7 @@ test_pool_max_op
 test_prelu_op
 test_prior_box_op
 test_psroi_pool_op
+test_put_along_axis_op
 test_range
 test_reverse_op
 test_roi_align_op