【PIR OpTest Fix No.23】 fix test_distribute_fpn_proposals_op (#60335)

* fix * fix
PaddlePaddle · Jan 2, 2024 · b56d140 · b56d140
1 parent a08580e
commit b56d140
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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