diff --git a/paddle/fluid/pir/dialect/operator/ir/ops.yaml b/paddle/fluid/pir/dialect/operator/ir/ops.yaml index b926b055daa6a..b992c139b8543 100644 --- a/paddle/fluid/pir/dialect/operator/ir/ops.yaml +++ b/paddle/fluid/pir/dialect/operator/ir/ops.yaml @@ -361,7 +361,7 @@ kernel : func : distribute_fpn_proposals data_type : fpn_rois - optional : rois_num + optional : rois_num, multi_level_rois_num - op : divide args : (Tensor x, Tensor y) diff --git a/test/white_list/pir_op_test_white_list b/test/white_list/pir_op_test_white_list index 2bf69d7d82faf..a0db4b9752069 100644 --- a/test/white_list/pir_op_test_white_list +++ b/test/white_list/pir_op_test_white_list @@ -80,6 +80,7 @@ test_diagonal_op test_digamma_op test_dirichlet_op test_dist_op +test_distribute_fpn_proposals_op test_dot_op test_dpsgd_op test_edit_distance_op