@@ -49,8 +49,12 @@ if(WIN32)
4949 "test_trt_convert_quantize_dequantize_linear" )
5050 list (REMOVE_ITEM TEST_TRT_IR_PASSES "test_trt_explicit_quantization" )
5151 list (REMOVE_ITEM TEST_TRT_IR_PASSES "test_trt_explicit_quantization_resnet" )
52+ list (REMOVE_ITEM TEST_INFERENCE_IR_PASSES
53+ "test_trt_explicit_quantization_resnet" )
5254 list (REMOVE_ITEM TEST_TRT_IR_PASSES
5355 "test_trt_explicit_quantization_mobilenet" )
56+ list (REMOVE_ITEM TEST_INFERENCE_IR_PASSES
57+ "test_trt_explicit_quantization_mobilenet" )
5458endif ()
5559
5660# Only for cpu(mkl + openblas)
@@ -110,7 +114,7 @@ foreach(TEST_INFERENCE_IR_PASS ${TEST_ONEDNN_IR_PASSES})
110114 list (REMOVE_ITEM TEST_INFERENCE_IR_PASSES ${TEST_INFERENCE_IR_PASS} )
111115endforeach ()
112116
113- if (WITH_MKLDNN)
117+ if (WITH_MKLDNN AND ( NOT WIN32 OR WIN_SPEED_UP_TEST_LEVEL EQUAL 0) )
114118 foreach (target ${TEST_MKLDNN_IR_PASSES} )
115119 py_test_modules (${target} MODULES ${target} )
116120 set_tests_properties (${target} PROPERTIES LABELS "RUN_TYPE=INFER" )
@@ -175,8 +179,10 @@ if(WITH_GPU AND TENSORRT_FOUND)
175179 PROPERTIES TIMEOUT 300 )
176180 set_tests_properties (test_trt_explicit_quantization_mobilenet
177181 PROPERTIES TIMEOUT 300 )
182+ else ()
183+ set_tests_properties (test_trt_convert_fill_constant PROPERTIES TIMEOUT 450 )
178184 endif ()
179- if (WITH_MKLDNN)
185+ if (WITH_MKLDNN AND ( NOT WIN32 OR WIN_SPEED_UP_TEST_LEVEL EQUAL 0) )
180186 set_tests_properties (test_save_optimized_model_pass PROPERTIES TIMEOUT 300 )
181187 endif ()
182188
@@ -231,12 +237,16 @@ if(WITH_GPU AND TENSORRT_FOUND)
231237 PROPERTIES TIMEOUT 120 )
232238 set_tests_properties (test_conv_elementwise_add_act_fuse_pass
233239 PROPERTIES TIMEOUT 120 )
234- set_tests_properties (test_onednn_conv_bias_fuse_pass PROPERTIES TIMEOUT 300 )
235- set_tests_properties (test_onednn_conv_concat_activation_fuse_pass
236- PROPERTIES TIMEOUT 300 )
237- set_tests_properties (test_onednn_multi_gru_fuse_pass PROPERTIES TIMEOUT 120 )
238- set_tests_properties (test_onednn_multi_gru_seq_fuse_pass PROPERTIES TIMEOUT
239- 120 )
240+ if (NOT WIN32 OR WIN_SPEED_UP_TEST_LEVEL EQUAL 0)
241+ set_tests_properties (test_onednn_conv_bias_fuse_pass PROPERTIES TIMEOUT
242+ 300 )
243+ set_tests_properties (test_onednn_conv_concat_activation_fuse_pass
244+ PROPERTIES TIMEOUT 300 )
245+ set_tests_properties (test_onednn_multi_gru_fuse_pass PROPERTIES TIMEOUT
246+ 120 )
247+ set_tests_properties (test_onednn_multi_gru_seq_fuse_pass
248+ PROPERTIES TIMEOUT 120 )
249+ endif ()
240250 set_tests_properties (test_flatten2_matmul_fuse_pass PROPERTIES TIMEOUT 240 )
241251 set_tests_properties (test_squeeze2_matmul_fuse_pass PROPERTIES TIMEOUT 240 )
242252 set_tests_properties (test_reshape2_matmul_fuse_pass PROPERTIES TIMEOUT 240 )
@@ -275,38 +285,43 @@ if(WITH_GPU AND TENSORRT_FOUND)
275285 endif ()
276286
277287 if (WITH_MKLDNN)
278- set_tests_properties (test_onednn_conv_bn_fuse_pass PROPERTIES TIMEOUT 120 )
279- set_tests_properties (test_onednn_conv_elementwise_add_fuse_pass
280- PROPERTIES TIMEOUT 120 )
281- set_tests_properties (test_mkldnn_depthwise_conv_pass PROPERTIES TIMEOUT 120 )
282- set_tests_properties (test_onednn_reshape_transpose_matmul_fuse_pass
283- PROPERTIES TIMEOUT 100 )
284- set_tests_properties (test_mkldnn_mish_op PROPERTIES TIMEOUT 300 )
285- set_tests_properties (test_mkldnn_conv3d_op PROPERTIES TIMEOUT 300 )
286- set_tests_properties (test_mkldnn_prelu_op PROPERTIES TIMEOUT 300 )
287288 set_tests_properties (test_conv_act_onednn_fuse_pass PROPERTIES TIMEOUT 120 )
288289 set_tests_properties (test_conv_transpose_eltwiseadd_bn_fuse_pass
289290 PROPERTIES TIMEOUT 250 )
290- set_tests_properties (test_onednn_matmul_transpose_reshape_fuse_pass
291- PROPERTIES TIMEOUT 100 )
292- set_tests_properties (test_conv_transpose_bn_fuse_pass PROPERTIES TIMEOUT
293- 300 )
294- set_tests_properties (test_mkldnn_conv_hard_sigmoid_fuse_pass
295- PROPERTIES TIMEOUT 300 )
296- set_tests_properties (test_mkldnn_conv_hard_swish_fuse_pass
297- PROPERTIES TIMEOUT 300 )
298- set_tests_properties (test_onednn_batch_norm_act_fuse_pass PROPERTIES TIMEOUT
299- 100 )
300- set_tests_properties (test_mkldnn_matmul_v2_transpose_reshape_fuse_pass
301- PROPERTIES TIMEOUT 100 )
302- set_tests_properties (test_mkldnn_conv_transpose_bias_fuse_pass
303- PROPERTIES TIMEOUT 100 )
304291 set_tests_properties (test_conv_eltwiseadd_bn_fuse_pass PROPERTIES TIMEOUT
305292 300 )
306- set_tests_properties (test_mkldnn_conv_mish_fuse_pass PROPERTIES TIMEOUT 300 )
307- set_tests_properties (test_onednn_fc_activation_fuse_pass PROPERTIES TIMEOUT
308- 300 )
309- set_tests_properties (test_mkldnn_conv_affine_channel_fuse_pass
310- PROPERTIES TIMEOUT 60 )
293+ if (NOT WIN32 OR WIN_SPEED_UP_TEST_LEVEL EQUAL 0)
294+ set_tests_properties (test_onednn_conv_bn_fuse_pass PROPERTIES TIMEOUT 120 )
295+ set_tests_properties (test_onednn_conv_elementwise_add_fuse_pass
296+ PROPERTIES TIMEOUT 120 )
297+ set_tests_properties (test_mkldnn_depthwise_conv_pass PROPERTIES TIMEOUT
298+ 120 )
299+ set_tests_properties (test_onednn_reshape_transpose_matmul_fuse_pass
300+ PROPERTIES TIMEOUT 100 )
301+ set_tests_properties (test_mkldnn_mish_op PROPERTIES TIMEOUT 300 )
302+ set_tests_properties (test_mkldnn_conv3d_op PROPERTIES TIMEOUT 300 )
303+ set_tests_properties (test_mkldnn_prelu_op PROPERTIES TIMEOUT 300 )
304+
305+ set_tests_properties (test_onednn_matmul_transpose_reshape_fuse_pass
306+ PROPERTIES TIMEOUT 100 )
307+ set_tests_properties (test_conv_transpose_bn_fuse_pass PROPERTIES TIMEOUT
308+ 300 )
309+ set_tests_properties (test_mkldnn_conv_hard_sigmoid_fuse_pass
310+ PROPERTIES TIMEOUT 300 )
311+ set_tests_properties (test_mkldnn_conv_hard_swish_fuse_pass
312+ PROPERTIES TIMEOUT 300 )
313+ set_tests_properties (test_onednn_batch_norm_act_fuse_pass
314+ PROPERTIES TIMEOUT 100 )
315+ set_tests_properties (test_mkldnn_matmul_v2_transpose_reshape_fuse_pass
316+ PROPERTIES TIMEOUT 100 )
317+ set_tests_properties (test_mkldnn_conv_transpose_bias_fuse_pass
318+ PROPERTIES TIMEOUT 100 )
319+ set_tests_properties (test_mkldnn_conv_mish_fuse_pass PROPERTIES TIMEOUT
320+ 300 )
321+ set_tests_properties (test_onednn_fc_activation_fuse_pass
322+ PROPERTIES TIMEOUT 300 )
323+ set_tests_properties (test_mkldnn_conv_affine_channel_fuse_pass
324+ PROPERTIES TIMEOUT 60 )
325+ endif ()
311326 endif ()
312327endif ()
0 commit comments