From 5a69dbcb87cf20211077248ce987a49299d60df0 Mon Sep 17 00:00:00 2001 From: kangguangli Date: Wed, 19 Jul 2023 19:45:46 +0800 Subject: [PATCH] fix put_along_axis (#55513) --- paddle/phi/api/yaml/backward.yaml | 4 ++-- test/white_list/new_ir_op_test_white_list | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/phi/api/yaml/backward.yaml b/paddle/phi/api/yaml/backward.yaml index 2ea1660e746..a109fac2158 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 40cfd7c2dfb..b17b87a0a58 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 -- GitLab