【PIR OpTest Fix No.23】 fix test_distribute_fpn_proposals_op (PaddlePa…

…ddle#60335) * fix * fix
Wanglongzhi2001 · Jan 7, 2024 · 043855d · 043855d
1 parent 217db4c
commit 043855d
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