set_pir_tests_properties for some tests (#60401)

xingmingyyj · web-flow · commit 5df9cdfdd9b8 · 2024-01-08T16:11:18.000+08:00
* fix

* Update CMakeLists.txt

* Update pir_op_test_white_list

* Update pir_op_test_white_list

* Update pir_op_test_white_list
diff --git a/test/auto_parallel/CMakeLists.txt b/test/auto_parallel/CMakeLists.txt
@@ -297,3 +297,5 @@ endif()
 
 py_test_modules(test_job_schedule_profiler_range MODULES
                 test_job_schedule_profiler_range)
+
+set_pit_tests_properties()
diff --git a/test/mkldnn/CMakeLists.txt b/test/mkldnn/CMakeLists.txt
@@ -31,3 +31,5 @@ if(WITH_MKLDNN AND NOT WIN32)
                        PROPERTIES TIMEOUT 300)
 endif()
 # set_tests_properties(test_flags_mkldnn_ops_on_off PROPERTIES TIMEOUT 120)
+
+set_pit_tests_properties()
diff --git a/test/white_list/pir_op_test_white_list b/test/white_list/pir_op_test_white_list
@@ -1,6 +1,5 @@
 test_accuracy_op
 test_activation_bf16_mkldnn_op
-test_activation_mkldnn_op
 test_activation_op
 test_adadelta_op
 test_adagrad_op
@@ -23,7 +22,6 @@ test_auc_single_pred_op
 test_bce_loss
 test_bernoulli_op               
 test_bicubic_interp_v2_op
-test_bilinear_interp_v2_mkldnn_op
 test_bilinear_interp_v2_op
 test_bilinear_tensor_product_op
 test_bincount_op
@@ -34,7 +32,6 @@ test_box_coder_op
 test_broadcast_error
 test_broadcast_tensors_op
 test_c_embedding_op
-test_cast_mkldnn_op
 test_cast_op
 test_channel_shuffle
 test_cholesky_op
@@ -48,20 +45,13 @@ test_compare_reduce_op
 test_complex_abs
 test_complex_op
 test_complex_view_op
-test_concat_bf16_mkldnn_op
 test_concat_int8_mkldnn_op
 test_concat_mkldnn_op
 test_concat_op
 test_conj_op
-test_conv2d_bf16_mkldnn_op
-test_conv2d_int8_mkldnn_op
-test_conv2d_mkldnn_op
 test_conv2d_op
-test_conv2d_transpose_bf16_mkldnn_op
-test_conv2d_transpose_mkldnn_op
 test_conv2d_transpose_op
 test_conv2d_transpose_op_depthwise_conv
-test_conv3d_mkldnn_op
 test_conv3d_op
 test_conv3d_transpose_part2_op
 test_crop_tensor_op
@@ -72,7 +62,6 @@ test_cumprod_op
 test_cumsum_op
 test_decayed_adagrad_op
 test_deformable_conv_op
-test_dequantize_mkldnn_op
 test_determinant_op
 test_diag_embed
 test_diag_v2
@@ -91,15 +80,12 @@ test_eigh_op_static_build
 test_eigvals_op
 test_eigvalsh_op
 test_einsum_op
-test_elementwise_add_bf16_mkldnn_op
 test_elementwise_div_op
 test_elementwise_floordiv_op
 test_elementwise_heaviside_op
 test_elementwise_max_op
 test_elementwise_min_op
 test_elementwise_mod_op
-test_elementwise_mul_bf16_mkldnn_op
-test_elementwise_mul_onednn_op
 test_elementwise_mul_op
 test_elementwise_pow_op
 test_erf_op
@@ -108,11 +94,8 @@ test_expand_as_v2_op
 test_expand_v2_op
 test_exponential_op
 test_eye_op
-test_fc_bf16_mkldnn_op
-test_fc_mkldnn_op
 test_fill_any_like_op
 test_fill_any_op
-test_fill_constant_batch_size_like
 test_fill_constant_op
 test_fill_diagonal_tensor_op
 test_flatten_contiguous_range_op
@@ -130,12 +113,6 @@ test_fused_fc_elementwise_layernorm_op
 test_fused_feedforward_op
 test_fused_gate_attention_op
 test_fused_multihead_matmul_op
-test_fusion_gru_bf16_mkldnn_op
-test_fusion_gru_int8_mkldnn_op
-test_fusion_gru_mkldnn_op
-test_fusion_lstm_bf16_mkldnn_op
-test_fusion_lstm_int8_mkldnn_op
-test_fusion_lstm_mkldnn_op
 test_fusion_seqexpand_concat_fc_op
 test_fusion_transpose_flatten_concat_op
 test_gather_nd_op
@@ -176,21 +153,17 @@ test_linear_interp_v2_op
 test_linspace
 test_log_loss_op
 test_log_softmax
-test_log_softmax_mkldnn_op
 test_logcumsumexp_op
 test_logit_op
 test_logspace
 test_logsumexp
 test_lookup_table_v2_bf16_op
 test_lookup_table_v2_op
 test_lookup_table_v2_op_static_build
-test_lrn_mkldnn_op
 test_lu_op
 test_lu_unpack_op
 test_margin_cross_entropy_op
 test_masked_select_op
-test_matmul_bf16_mkldnn_op
-test_matmul_mkldnn_op
 test_matmul_v2_op
 test_matmul_v2_op_static_build
 test_matrix_nms_op
@@ -202,17 +175,14 @@ test_memcpy_op
 test_meshgrid_op
 test_mode_op
 test_momentum_op
-test_mul_int8_mkldnn_op
 test_mul_op
 test_multi_dot_op
 test_multi_forward
-test_multi_gru_mkldnn_op
 test_multinomial_op             
 test_multiplex_op
 test_mv_op
 test_nanmedian
 test_nce
-test_nearest_interp_mkldnn_op
 test_nearest_interp_v2_op
 test_nextafter_op
 test_nll_loss
@@ -230,8 +200,6 @@ test_pass_quantization
 test_pixel_shuffle_op
 test_poisson_op                 
 test_polygamma_op
-test_pool2d_int8_mkldnn_op
-test_pool2d_mkldnn_op
 test_pool2d_op
 test_pool_max_op
 test_prelu_mkldnn_op
@@ -240,25 +208,20 @@ test_prior_box_op
 test_psroi_pool_op
 test_put_along_axis_op
 test_qr_op
-test_quantize_mkldnn_op
 test_randint_op
 test_randperm_op                
 test_range
-test_reduce_mkldnn_op
 test_reduce_op
 test_reduce_op_static_build
 test_repeat_interleave_op
-test_requantize_mkldnn_op
 test_reshape_bf16_op
-test_reshape_mkldnn_op
 test_reshape_op
 test_reverse_op
 test_roi_align_op
 test_roi_pool_op
 test_roll_op
 test_row_conv_op
 test_rrelu_op
-test_scale_mkldnn_op
 test_scale_op
 test_scatter_nd_op
 test_scatter_op
@@ -276,9 +239,7 @@ test_sigmoid_cross_entropy_with_logits_op
 test_sign_op
 test_size_op
 test_slice_op
-test_softmax_bf16_mkldnn_op
 test_softmax_mask_fuse_upper_triangle_op
-test_softmax_mkldnn_op
 test_softmax_op
 test_solve_op
 test_sparse_momentum_op
@@ -288,7 +249,6 @@ test_split_mkldnn_op
 test_split_op
 test_squared_l2_norm_op
 test_squeeze2_op
-test_sum_bf16_mkldnn_op
 test_sum_mkldnn_op
 test_svd_op
 test_take_along_axis_op
