diff --git a/paddle/fluid/pir/dialect/op_generator/vjp_interface_black_list.py b/paddle/fluid/pir/dialect/op_generator/vjp_interface_black_list.py index 6f588aa7186191..2bf56de2a5859c 100644 --- a/paddle/fluid/pir/dialect/op_generator/vjp_interface_black_list.py +++ b/paddle/fluid/pir/dialect/op_generator/vjp_interface_black_list.py @@ -40,4 +40,5 @@ 'acos_grad', 'put_along_axis_grad', 'masked_fill_grad', + 'masked_select_grad', ] diff --git a/paddle/phi/ops/yaml/backward.yaml b/paddle/phi/ops/yaml/backward.yaml index 39503368a7a5ae..391ee7bbcecddf 100644 --- a/paddle/phi/ops/yaml/backward.yaml +++ b/paddle/phi/ops/yaml/backward.yaml @@ -2164,6 +2164,12 @@ no_need_buffer : x, value backward: masked_fill_double_grad +- backward_op : masked_select_double_grad + forward: masked_select_grad (Tensor x, Tensor mask, Tensor grad_out) -> Tensor(grad_x) + args : (Tensor mask, Tensor grad_x_grad) + output : Tensor(grad_out_grad) + invoke : masked_select(grad_x_grad, mask) + - backward_op : masked_select_grad forward : masked_select (Tensor x, Tensor mask) -> Tensor(out) args : (Tensor x, Tensor mask, Tensor out_grad) @@ -2175,6 +2181,7 @@ func : masked_select_grad data_type : x no_need_buffer : x + backward: masked_select_double_grad - backward_op : match_matrix_tensor_grad forward: match_matrix_tensor(Tensor x, Tensor y, Tensor w, int dim_t = 1) -> Tensor (out), Tensor (tmp) diff --git a/paddle/phi/ops/yaml/op_compat.yaml b/paddle/phi/ops/yaml/op_compat.yaml index 150e9f0f1e35df..bd49cac4678f3f 100755 --- a/paddle/phi/ops/yaml/op_compat.yaml +++ b/paddle/phi/ops/yaml/op_compat.yaml @@ -2443,6 +2443,7 @@ {softmax : Softmax, loss : Loss} - op : masked_select + backward : masked_select_grad, masked_select_double_grad inputs : {x : X, mask : Mask} outputs :