diff --git a/tools/parallel_UT_rule.py b/tools/parallel_UT_rule.py index 96ffa5ee860c78d0aab75f87ad618992be8ba500..647c36c8954e0749166ef3e7585239c8fb9e37ac 100644 --- a/tools/parallel_UT_rule.py +++ b/tools/parallel_UT_rule.py @@ -1474,14 +1474,11 @@ CPU_PARALLEL_JOB = [ 'test_create_op_doc_string', 'test_create_global_var', 'test_cpu_quantize_placement_pass', - 'test_cpu_quantize_pass', 'test_cpu_bfloat16_placement_pass', 'test_cpu_bfloat16_pass', - 'test_conv_elementwise_add_mkldnn_fuse_pass', 'test_conv_concat_relu_mkldnn_fuse_pass', 'test_conv_bias_mkldnn_fuse_pass', 'test_conv_batch_norm_mkldnn_fuse_pass', - 'test_conv_activation_mkldnn_fuse_pass', 'test_conv3d_transpose_layer', 'test_conv3d_mkldnn_op', 'test_conv3d_layer', @@ -1519,7 +1516,6 @@ CPU_PARALLEL_JOB = [ 'stringprintf_test', 'stringpiece_test', 'split_test', - 'selected_rows_test', 'selected_rows_functor_test', 'scope_test', 'scatter_test', @@ -1555,7 +1551,6 @@ CPU_PARALLEL_JOB = [ 'lod_tensor_test', 'test_check_abi', 'lodtensor_printer_test', - 'jit_kernel_test', 'test_dispatch_jit', 'inlined_vector_test', 'infer_io_utils_tester', @@ -1803,7 +1798,6 @@ TETRAD_PARALLEL_JOB = [ 'test_analyzer_bert', 'test_analyzer_googlenet', 'test_fleet_base', - 'test_imperative_layers', 'test_dgc_momentum_op', 'test_memcpy_op', 'test_dgc_op', @@ -1815,9 +1809,13 @@ TETRAD_PARALLEL_JOB = [ 'test_collective_split_embedding', 'float16_gpu_test', 'test_leaky_relu_grad_grad_functor', - 'test_complex_simplenet', 'selected_rows_functor_gpu_test', 'test_imperative_framework', + 'selected_rows_test', + 'test_conv_elementwise_add_mkldnn_fuse_pass', + 'test_cpu_quantize_pass', + 'jit_kernel_test', + 'test_conv_activation_mkldnn_fuse_pass', ] # It run 2 job each time, If it failed due to Insufficient GPU memory or CUBLAS_STATUS_ALLOC_FAILED, @@ -2191,6 +2189,8 @@ TWO_PARALLEL_JOB = [ 'test_analyzer_ernie', 'lite_resnet50_test', 'lite_mul_model_test', + 'test_complex_simplenet', + 'test_imperative_layers', ] diff --git a/tools/windows/run_unittests.sh b/tools/windows/run_unittests.sh index 86f5d892baa017a5bf418dcecef63e4f2ff37e5f..dcf73189a773234a0bd3737c6221e64efbb8e77e 100644 --- a/tools/windows/run_unittests.sh +++ b/tools/windows/run_unittests.sh @@ -155,7 +155,6 @@ long_time_test="^test_gru_op$|\ ^test_strided_slice_op$" if [ ${WITH_GPU:-OFF} == "ON" ];then - export FLAGS_fraction_of_gpu_memory_to_use=0.92 export CUDA_VISIBLE_DEVICES=0 UT_list=$(ctest -N | awk -F ': ' '{print $2}' | sed '/^$/d' | sed '$d')