CMakeLists.txt 67.4 KB
Newer Older
1 2
file(GLOB TEST_OPS RELATIVE
"${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
3
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
4
set(GC_ENVS FLAGS_eager_delete_tensor_gb=0.0 FLAGS_fast_eager_deletion_mode=1 FLAGS_memory_fraction_of_eager_deletion=1.0)
5
set(dist_ENVS http_proxy="" https_proxy="")
6

7
file(GLOB DIST_TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_dist_*.py")
8
list(REMOVE_ITEM DIST_TEST_OPS "test_dist_op")
9
if ((NOT WITH_NCCL) AND (NOT WITH_RCCL))
10 11
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_mnist_dgc_nccl")
endif()
12

13
string(REPLACE ".py" "" DIST_TEST_OPS "${DIST_TEST_OPS}")
14 15 16 17 18 19 20 21

if ((NOT WITH_GPU) AND (NOT WITH_XPU) AND NOT (WITH_ASCEND OR WITH_ASCEND_CL))
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_fleet_grad_clip")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_fleet_heter_ctr")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_fleet_ps_gpu_ctr")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_mnist_batch_merge")
endif()

22
list(APPEND DIST_TEST_OPS test_parallel_dygraph_mnist)
23
list(APPEND DIST_TEST_OPS test_pipeline)
24
list(APPEND DIST_TEST_OPS test_ir_pass_pipeline)
L
lilong12 已提交
25
list(APPEND DIST_TEST_OPS test_static_model_parallel)
26
list(APPEND DIST_TEST_OPS test_static_model_parallel_fused_feedforward)
27
list(APPEND DIST_TEST_OPS test_static_model_parallel_fused_attention)
28
list(APPEND DIST_TEST_OPS test_static_model_parallel_fused_multi_transformer)
29
list(APPEND DIST_TEST_OPS test_parallel_dygraph_se_resnext)
30
list(APPEND DIST_TEST_OPS test_parallel_dygraph_sparse_embedding)
31
list(APPEND DIST_TEST_OPS test_parallel_dygraph_sparse_embedding_over_height)
32
list(APPEND DIST_TEST_OPS test_parallel_dygraph_transformer)
33 34 35 36 37 38 39 40
if (WITH_GPU OR WITH_XPU OR WITH_ASCEND OR WITH_ASCEND_CL)
    list(APPEND DIST_TEST_OPS test_fleet_graph_execution_meta_optimizer)
    list(APPEND DIST_TEST_OPS test_fleet_pipeline_meta_optimizer)
    list(APPEND DIST_TEST_OPS test_fleet_pipeline_meta_optimizer_with_recompute)
    list(APPEND DIST_TEST_OPS test_fleet_raw_program_meta_optimizer)
    list(APPEND DIST_TEST_OPS test_gen_nccl_id_op)
    list(APPEND DIST_TEST_OPS test_rnn_dp)
endif()
41
list(APPEND DIST_TEST_OPS test_parallel_dygraph_unused_variables)
42
list(APPEND DIST_TEST_OPS test_parallel_dygraph_control_flow)
43 44
list(APPEND DIST_TEST_OPS test_parallel_dygraph_no_sync)
list(APPEND DIST_TEST_OPS test_parallel_dygraph_no_sync_gradient_check)
45
list(APPEND DIST_TEST_OPS test_parallel_dygraph_dataparallel)
46 47
list(APPEND DIST_TEST_OPS test_parallel_dygraph_pipeline_parallel)
list(APPEND DIST_TEST_OPS test_parallel_dygraph_tensor_parallel)
J
JZ-LIANG 已提交
48
list(APPEND DIST_TEST_OPS test_parallel_dygraph_sharding_parallel)
49
list(APPEND DIST_TEST_OPS test_dygraph_sharding_optimizer_stage2)
B
Baibaifan 已提交
50
list(APPEND DIST_TEST_OPS test_dygraph_sharding_stage2)
B
Baibaifan 已提交
51
list(APPEND DIST_TEST_OPS test_dygraph_sharding_stage3)
B
Baibaifan 已提交
52
list(APPEND DIST_TEST_OPS test_dygraph_group_sharded_api)
53
list(APPEND DIST_TEST_OPS test_auto_parallel_parallelizer)
54
list(APPEND DIST_TEST_OPS test_parallel_dygraph_mp_layers)
55
list(APPEND DIST_TEST_OPS test_hybrid_parallel_inference_helper)
56
list(APPEND DIST_TEST_OPS test_parallel_class_center_sample)
57
list(APPEND DIST_TEST_OPS test_parallel_margin_cross_entropy)
58
list(APPEND DIST_TEST_OPS test_auto_parallel_data_unshard)
59
list(APPEND DIST_TEST_OPS test_auto_parallel_save_load)
60
list(APPEND DIST_TEST_OPS test_auto_parallel_autoconvert)
61
list(APPEND DIST_TEST_OPS test_collective_process_group)
62
list(APPEND DIST_TEST_OPS test_eager_dist_api)
63 64 65
set(MIXED_DIST_TEST_OPS ${DIST_TEST_OPS})
#remove distribute unittests.
list(APPEND MIXED_DIST_TEST_OPS test_dgc_op)
66 67
list(APPEND MIXED_DIST_TEST_OPS test_dgc_momentum_op)
list(APPEND MIXED_DIST_TEST_OPS test_dgc_optimizer)
68
list(APPEND MIXED_DIST_TEST_OPS test_simple_dist_transpiler)
T
tangwei12 已提交
69
list(APPEND MIXED_DIST_TEST_OPS test_recv_save_op)
70
list(APPEND MIXED_DIST_TEST_OPS test_c_comm_init_op)
71
list(APPEND MIXED_DIST_TEST_OPS test_communicator_async)
T
Thunderbrook 已提交
72
list(APPEND MIXED_DIST_TEST_OPS test_communicator_ps_gpu)
73 74
list(APPEND MIXED_DIST_TEST_OPS test_communicator_geo)
list(APPEND MIXED_DIST_TEST_OPS test_communicator_half_async)
T
tangwei12 已提交
75
list(APPEND MIXED_DIST_TEST_OPS test_communicator_sync)
76 77 78 79 80 81
list(APPEND MIXED_DIST_TEST_OPS test_fleet_launch_ps)
list(APPEND MIXED_DIST_TEST_OPS test_launch_coverage)
list(APPEND MIXED_DIST_TEST_OPS test_fleetrun)
list(APPEND MIXED_DIST_TEST_OPS test_fleet_run_random_port)
list(APPEND MIXED_DIST_TEST_OPS test_fleet_launch_async)
list(APPEND MIXED_DIST_TEST_OPS test_fleet_launch_cloud)
82 83
list(APPEND MIXED_DIST_TEST_OPS test_fleet_launch_ascend)
list(APPEND MIXED_DIST_TEST_OPS test_ascend_group)
84
list(APPEND MIXED_DIST_TEST_OPS test_fleet_launch_nproc)
85
list(APPEND MIXED_DIST_TEST_OPS test_fleet_api_input)
M
mapingshuo 已提交
86
list(APPEND MIXED_DIST_TEST_OPS test_collective_optimizer)
87
list(APPEND MIXED_DIST_TEST_OPS test_fleet_base)
88 89
list(APPEND MIXED_DIST_TEST_OPS test_fleet_base_2)
list(APPEND MIXED_DIST_TEST_OPS test_fleet_base_3)
90
list(APPEND MIXED_DIST_TEST_OPS test_fleet_recompute_meta_optimizer)
91
list(APPEND MIXED_DIST_TEST_OPS test_fleet_pipeline_meta_optimizer)
92
list(APPEND MIXED_DIST_TEST_OPS test_fleet_pipeline_meta_optimizer_with_recompute)
93
list(APPEND MIXED_DIST_TEST_OPS test_fleet_raw_program_meta_optimizer)
李季 已提交
94
list(APPEND MIXED_DIST_TEST_OPS test_rnn_dp)
95
list(APPEND MIXED_DIST_TEST_OPS test_fleet_amp_meta_optimizer)
96
list(APPEND MIXED_DIST_TEST_OPS test_fleet_amp_init)
97
list(APPEND MIXED_DIST_TEST_OPS test_fleet_gradient_merge_meta_optimizer)
98
list(APPEND MIXED_DIST_TEST_OPS test_fleet_sharding_meta_optimizer)
99
list(APPEND MIXED_DIST_TEST_OPS test_fleet_hybrid_meta_optimizer)
Y
Yi Liu 已提交
100
list(APPEND MIXED_DIST_TEST_OPS test_fleet_localsgd_meta_optimizer)
101
list(APPEND MIXED_DIST_TEST_OPS test_fleet_lars_meta_optimizer)
102
list(APPEND MIXED_DIST_TEST_OPS test_fleet_lamb_meta_optimizer)
103
list(APPEND MIXED_DIST_TEST_OPS test_fleet_dgc_meta_optimizer)
104
list(APPEND MIXED_DIST_TEST_OPS test_fleet_fp16_allreduce_meta_optimizer)
105
list(APPEND MIXED_DIST_TEST_OPS test_fleet_private_function)
106
list(APPEND MIXED_DIST_TEST_OPS test_fleet_graph_executor)
107
list(APPEND MIXED_DIST_TEST_OPS test_fleet_meta_optimizer_base)
108
list(APPEND MIXED_DIST_TEST_OPS test_fleet_distributed_strategy)
109
list(APPEND MIXED_DIST_TEST_OPS test_fleet_auto)
W
WangXi 已提交
110
list(APPEND MIXED_DIST_TEST_OPS test_fleet_static_mp_layers)
111 112
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_partitioner)
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_partitioner_gpt)
C
caozhou 已提交
113
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_searcher)
C
caozhou 已提交
114
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_reshard)
115
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_dist_tensor)
C
caozhou 已提交
116 117 118
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_reshard_serial)
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_reshard_mppp)
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_reshard_dpmppp)
119
list(APPEND MIXED_DIST_TEST_OPS test_auto_parallel_cost_model)
120 121 122
foreach(TEST_OP ${MIXED_DIST_TEST_OPS})
  list(REMOVE_ITEM TEST_OPS ${TEST_OP})
endforeach()
123

124
if(NOT WITH_PYTHON AND ON_INFER)
125 126 127
    LIST(REMOVE_ITEM TEST_OPS test_eager_trace_op)
endif()

L
Li Min 已提交
128
if(NOT WITH_GPU)
129
    LIST(REMOVE_ITEM TEST_OPS test_fused_feedforward_op)
L
Li Min 已提交
130
    LIST(REMOVE_ITEM TEST_OPS test_fused_attention_op)
131
    LIST(REMOVE_ITEM TEST_OPS test_fused_attention_op_api)
132
    LIST(REMOVE_ITEM TEST_OPS test_fused_multi_transformer_op)
133
    LIST(REMOVE_ITEM TEST_OPS test_fused_transformer_encoder_layer)
134 135
endif()

136 137 138
LIST(REMOVE_ITEM TEST_OPS test_fused_gemm_epilogue_op)
LIST(REMOVE_ITEM TEST_OPS test_fused_gemm_epilogue_grad_op)
LIST(REMOVE_ITEM TEST_OPS test_fuse_gemm_epilogue_pass)
L
Li Min 已提交
139

140
if(((NOT WITH_ROCM) AND (NOT WITH_GPU)) OR WIN32)
141
    LIST(REMOVE_ITEM TEST_OPS test_c_comm_init_all_op)
L
lilong12 已提交
142 143
    LIST(REMOVE_ITEM TEST_OPS test_c_concat)
    LIST(REMOVE_ITEM TEST_OPS test_c_split)
144
    LIST(REMOVE_ITEM TEST_OPS test_allgather)
L
lilong12 已提交
145
    LIST(REMOVE_ITEM TEST_OPS test_c_identity)
146
    LIST(REMOVE_ITEM TEST_OPS test_c_embedding_op)
147 148
    LIST(REMOVE_ITEM TEST_OPS test_allreduce)
    LIST(REMOVE_ITEM TEST_OPS test_broadcast)
L
lilong12 已提交
149
    LIST(REMOVE_ITEM TEST_OPS test_collective_reduce)
150
    LIST(REMOVE_ITEM TEST_OPS test_pipeline_parallel)
L
lilong12 已提交
151
    LIST(REMOVE_ITEM TEST_OPS test_collective_scatter)
L
lilong12 已提交
152
    LIST(REMOVE_ITEM TEST_OPS test_collective_sendrecv)
153
    LIST(REMOVE_ITEM TEST_OPS test_reducescatter)
154
    LIST(REMOVE_ITEM TEST_OPS test_reducescatter_api)
155 156 157 158
    LIST(REMOVE_ITEM TEST_OPS test_collective_split_embedding)
    LIST(REMOVE_ITEM TEST_OPS test_collective_split_embedding_none_divisible)
    LIST(REMOVE_ITEM TEST_OPS test_collective_split_row_linear)
    LIST(REMOVE_ITEM TEST_OPS test_collective_split_col_linear)
L
lilong12 已提交
159 160 161 162
    LIST(REMOVE_ITEM TEST_OPS test_collective_reduce_api)
    LIST(REMOVE_ITEM TEST_OPS test_collective_scatter_api)
    LIST(REMOVE_ITEM TEST_OPS test_collective_barrier_api)
    LIST(REMOVE_ITEM TEST_OPS test_collective_allreduce_api)
K
kuizhiqing 已提交
163
    LIST(REMOVE_ITEM TEST_OPS test_new_group_api)
L
lilong12 已提交
164 165
    LIST(REMOVE_ITEM TEST_OPS test_collective_broadcast_api)
    LIST(REMOVE_ITEM TEST_OPS test_collective_allgather_api)
L
lilong12 已提交
166
    LIST(REMOVE_ITEM TEST_OPS test_collective_alltoall_api)
167 168
    LIST(REMOVE_ITEM TEST_OPS test_collective_global_gather)
    LIST(REMOVE_ITEM TEST_OPS test_collective_global_scatter)
L
lilong12 已提交
169
    LIST(REMOVE_ITEM TEST_OPS test_collective_sendrecv_api)
W
WangXi 已提交
170
    LIST(REMOVE_ITEM TEST_OPS test_collective_wait)
J
JZ-LIANG 已提交
171
    LIST(REMOVE_ITEM TEST_OPS test_memcpy_op)
李季 已提交
172
    LIST(REMOVE_ITEM TEST_OPS test_raw_program_optimizer)
Y
Yuang Liu 已提交
173
    LIST(REMOVE_ITEM TEST_OPS test_fleet_gradient_scale)
174
    LIST(REMOVE_ITEM TEST_OPS test_disable_signal_handler)
L
LiYuRio 已提交
175
    LIST(REMOVE_ITEM TEST_OPS test_fleet_executor)
176
    LIST(REMOVE_ITEM TEST_OPS test_fleet_executor_with_task_nodes)
177
    LIST(REMOVE_ITEM TEST_OPS test_fleet_executor_multi_devices)
178
    LIST(REMOVE_ITEM TEST_OPS test_fleet_executor_origin_scheduler)
179
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_mapper)
L
LiYuRio 已提交
180
    LIST(REMOVE_ITEM TEST_OPS test_fleet_executor_task_node)
181 182
    LIST(REMOVE_ITEM TEST_OPS test_fleet_exe_dist_model_run)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_exe_dist_model_tensor)
183 184
endif()

185 186 187
# Temporally disable test_deprecated_decorator
LIST(REMOVE_ITEM TEST_OPS test_deprecated_decorator)

188 189 190 191
if(WIN32)
    LIST(REMOVE_ITEM TEST_OPS test_multiprocess_reader_exception)
    LIST(REMOVE_ITEM TEST_OPS test_trainer_desc)
    LIST(REMOVE_ITEM TEST_OPS test_checkpoint_notify_op)
192 193 194 195 196 197 198
    LIST(REMOVE_ITEM TEST_OPS test_downpoursgd)
    LIST(REMOVE_ITEM TEST_OPS test_fleet)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_nocvm_1)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_rolemaker)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_rolemaker_3)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_unitaccessor)
    LIST(REMOVE_ITEM TEST_OPS test_ps_dispatcher)
199 200
    LIST(REMOVE_ITEM TEST_OPS test_ir_memory_optimize_nlp)
    LIST(REMOVE_ITEM TEST_OPS test_nvprof)
201

202 203
    # TODO: Fix these unittests failed on Windows
    LIST(REMOVE_ITEM TEST_OPS test_debugger)
H
huangxu96 已提交
204 205 206
    if (WITH_GPU)
        LIST(REMOVE_ITEM TEST_OPS test_update_loss_scaling_op)
    endif()
207 208 209 210 211 212 213 214 215 216
endif()

if(NOT WITH_DISTRIBUTE OR WIN32)
    # DISTRIBUTE related
    LIST(REMOVE_ITEM TEST_OPS test_avoid_twice_initialization)
    LIST(REMOVE_ITEM TEST_OPS test_distributed_strategy)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_metric)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_ps)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_rolemaker_2)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_utils)
217
    LIST(REMOVE_ITEM TEST_OPS test_collective_cpu_barrier_with_gloo)
218 219

    # TODO: Fix these unittests failed on Windows
220
    list(REMOVE_ITEM TEST_OPS test_fake_init_op)
H
hutuxian 已提交
221 222
endif()

223 224 225 226 227
if(NOT WITH_DISTRIBUTE)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_rolemaker_new)
    LIST(REMOVE_ITEM TEST_OPS test_desc_clone_dist)
endif()

228 229
if(WIN32)
    LIST(REMOVE_ITEM TEST_OPS test_complex_matmul)
230
    LIST(REMOVE_ITEM TEST_OPS test_ops_nms)
231 232
endif()

G
gongweibao 已提交
233
LIST(REMOVE_ITEM TEST_OPS test_fleet_checkpoint)
234
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint)
235
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint1)
236
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint2)
237 238 239 240 241 242
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint3)
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint_multiple)
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint_dist_basic)
LIST(REMOVE_ITEM TEST_OPS test_hdfs1)
LIST(REMOVE_ITEM TEST_OPS test_hdfs2)
LIST(REMOVE_ITEM TEST_OPS test_hdfs3)
243
LIST(REMOVE_ITEM TEST_OPS test_checkpoint_saver)
T
tangwei12 已提交
244

G
gongweibao 已提交
245 246
if(APPLE OR WIN32)
    LIST(REMOVE_ITEM TEST_OPS test_fs_interface)
247
    LIST(REMOVE_ITEM TEST_OPS test_fleet_metric)
G
gongweibao 已提交
248 249
endif()

S
ShenLiang 已提交
250 251
list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_hybrid_parallel)

X
xiongkun 已提交
252 253
LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_transformer_gloo) # NOTE: @xiongkun03, cpu is too slow, fix it in next PR

254 255
if (NOT WITH_GLOO)
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_dataparallel_cpuonly)
X
xiongkun 已提交
256 257 258 259

    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_unused_variables_gloo)
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_sparse_embedding_over_height_gloo)
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_sparse_embedding_gloo)
260
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_sparse_embedding_diff_length_gloo)
261 262
endif()

263
if ((NOT WITH_GPU) AND (NOT WITH_ROCM))
Y
Yu Yang 已提交
264
    LIST(REMOVE_ITEM TEST_OPS test_conv2d_fusion_op)
S
ShenLiang 已提交
265
    LIST(REMOVE_ITEM TEST_OPS test_rank_attention_op) # TODO(shenliang03): rank_attention_op support CPU device in future
S
ShenLiang 已提交
266
    LIST(REMOVE_ITEM TEST_OPS test_batch_fc_op) # TODO(shenliang03): batch_fc_op support CPU device in future
267
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_mnist) # TODO(Yancey1989): parallel dygraph support CPU device in future
268
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_unused_variables)
269 270
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_se_resnext)
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_sparse_embedding)
271
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_sparse_embedding_over_height)
272
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_transformer)
C
ceci3 已提交
273
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_sync_batch_norm)
274
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_control_flow)
275 276
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_no_sync)
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_no_sync_gradient_check)
277
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_dataparallel)
278 279
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_pipeline_parallel)
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_tensor_parallel)
J
JZ-LIANG 已提交
280
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_sharding_parallel)
281
    list(REMOVE_ITEM TEST_OPS test_dygraph_sharding_optimizer_stage2)
B
Baibaifan 已提交
282
    list(REMOVE_ITEM TEST_OPS test_dygraph_sharding_stage2)
B
Baibaifan 已提交
283
    list(REMOVE_ITEM TEST_OPS test_dygraph_sharding_stage3)
B
Baibaifan 已提交
284
    list(REMOVE_ITEM TEST_OPS test_dygraph_group_sharded_api)
285
    list(REMOVE_ITEM TEST_OPS test_auto_parallel_parallelizer)
286
    list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_mp_layers)
287
    LIST(REMOVE_ITEM TEST_OPS test_imperative_auto_mixed_precision)
288
    LIST(REMOVE_ITEM TEST_OPS test_mixed_precision)
289
    LIST(REMOVE_ITEM TEST_OPS test_fleet_base_single)
J
JZ-LIANG 已提交
290
    LIST(REMOVE_ITEM TEST_OPS test_dygraph_recompute)
291
    list(REMOVE_ITEM TEST_OPS test_hybrid_parallel_inference_helper)
292
    list(REMOVE_ITEM TEST_OPS test_parallel_class_center_sample)
293
    LIST(REMOVE_ITEM TEST_OPS test_parallel_margin_cross_entropy)
294 295
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_partitioner)
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_partitioner_gpt)
C
caozhou 已提交
296
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_searcher)
C
caozhou 已提交
297
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_reshard)
298
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_dist_tensor)
C
caozhou 已提交
299 300 301
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_reshard_serial)
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_reshard_mppp)
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_reshard_dpmppp)
302
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_cost_model)
303
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_data_unshard)
304
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_save_load)
305
    LIST(REMOVE_ITEM TEST_OPS test_auto_parallel_autoconvert)
306
    LIST(REMOVE_ITEM TEST_OPS test_collective_process_group)
307
    LIST(REMOVE_ITEM TEST_OPS test_eager_dist_api)
308 309 310 311
elseif(WITH_GPU)
    if (${CUDNN_VERSION} VERSION_LESS 7100)
        LIST(REMOVE_ITEM TEST_OPS test_conv2d_fusion_op)
    endif()
D
Dang Qingqing 已提交
312 313
endif()

314 315 316
if (WITH_NCCL)
    if (${NCCL_VERSION} VERSION_LESS 2212)
        LIST(REMOVE_ITEM DIST_TEST_OPS test_parallel_dygraph_sparse_embedding)
317
        LIST(REMOVE_ITEM DIST_TEST_OPS test_parallel_dygraph_sparse_embedding_over_height)
318 319 320 321
        LIST(REMOVE_ITEM DIST_TEST_OPS test_parallel_dygraph_transformer)
    endif()
endif()

322
if ((NOT WITH_NCCL) AND (NOT WITH_RCCL))
323
    list(REMOVE_ITEM TEST_OPS test_imperative_group)
K
kuizhiqing 已提交
324
    LIST(REMOVE_ITEM TEST_OPS test_new_group_api)
325 326
endif()

327
if(((NOT WITH_ROCM) AND (NOT WITH_GPU)) OR WIN32)
H
hutuxian 已提交
328
    LIST(REMOVE_ITEM TEST_OPS test_boxps)
329
endif()
330 331 332
list(REMOVE_ITEM TEST_OPS test_seq_concat_op) # FIXME(helin): https://github.com/PaddlePaddle/Paddle/issues/8290
list(REMOVE_ITEM TEST_OPS test_lstm_unit_op) # # FIXME(qijun) https://github.com/PaddlePaddle/Paddle/issues/5185
list(REMOVE_ITEM TEST_OPS test_cond_op) # FIXME(qijun): https://github.com/PaddlePaddle/Paddle/issues/5101#issuecomment-339814957
H
Hongyu Liu 已提交
333

334
list(REMOVE_ITEM TEST_OPS op_test) # op_test is a helper python file, not a test
335
list(REMOVE_ITEM TEST_OPS decorator_helper) # decorator_helper is a helper python file, not a test
336

337
if(APPLE)
338 339 340 341
    if(NOT WITH_DISTRIBUTE)
        list(REMOVE_ITEM TEST_OPS test_desc_clone)
        list(REMOVE_ITEM TEST_OPS test_program_code)
    endif(NOT WITH_DISTRIBUTE)
342
    message(WARNING "These tests has been disabled in OSX before being fixed:\n test_fuse_elewise_add_act_pass \n test_detection_map_op \n test_dist_se_resnext_*")
343 344
    # this op is not support on mac
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
345
    list(REMOVE_ITEM TEST_OPS test_detection_map_op)
346
    list(REMOVE_ITEM TEST_OPS test_fuse_elewise_add_act_pass)
347
endif()
T
Tao Luo 已提交
348 349 350 351
if(NOT WITH_MKLML)
    # this op is not support on openblas
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
endif()
352

353
if(NOT WITH_MKL OR NOT WITH_AVX)
A
Aurelius84 已提交
354
  list(REMOVE_ITEM TEST_OPS test_match_matrix_tensor_op)
K
Kevin 已提交
355
  list(REMOVE_ITEM TEST_OPS test_var_conv_2d)
356
endif()
357

358
if(WITH_COVERAGE OR WIN32 OR WITH_NV_JETSON)
A
Aurelius84 已提交
359
  list(REMOVE_ITEM TEST_OPS test_pyramid_hash_op)
360 361
endif()

T
tangwei12 已提交
362
list(REMOVE_ITEM TEST_OPS test_fleet_pyramid_hash)
K
Kevin 已提交
363

364
if((WITH_ROCM OR WITH_GPU) OR NOT WITH_MKLML)
365 366 367 368
    # matmul with multiple heads need MKL support
    LIST(REMOVE_ITEM TEST_OPS test_matmul_op_with_head)
endif()

Y
Yanghello 已提交
369 370 371 372
if(NOT WITH_CRYPTO)
    LIST(REMOVE_ITEM TEST_OPS test_crypto)
endif()

373 374
function(py_test_modules TARGET_NAME)
  if(WITH_TESTING)
375
    set(options SERIAL)
376
    set(oneValueArgs "")
377
    set(multiValueArgs MODULES DEPS ENVS)
378
    cmake_parse_arguments(py_test_modules "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
379

380
    if(WITH_COVERAGE AND NOT (WITH_INCREMENTAL_COVERAGE AND "$ENV{PADDLE_GIT_DIFF_PY_FILE}" STREQUAL ""))
381 382 383 384 385 386 387 388 389 390 391 392 393
        if(WITH_ASCEND_CL)
            add_test(NAME ${TARGET_NAME}
                    COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python:$ENV{PYTHONPATH} ${py_test_modules_ENVS}
                    COVERAGE_FILE=${PADDLE_BINARY_DIR}/python-coverage.data
                    ${PYTHON_EXECUTABLE} -m coverage run --branch -p ${PADDLE_SOURCE_DIR}/tools/test_runner.py ${py_test_modules_MODULES}
                    WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
        else()
            add_test(NAME ${TARGET_NAME}
                    COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python ${py_test_modules_ENVS}
                    COVERAGE_FILE=${PADDLE_BINARY_DIR}/python-coverage.data
                    ${PYTHON_EXECUTABLE} -m coverage run --branch -p ${PADDLE_SOURCE_DIR}/tools/test_runner.py ${py_test_modules_MODULES}
                    WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
        endif()
394
    else()
395 396 397 398 399 400 401 402 403 404 405
        if(WITH_ASCEND_CL)
            add_test(NAME ${TARGET_NAME}
                    COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python:$ENV{PYTHONPATH} ${py_test_modules_ENVS}
                    ${PYTHON_EXECUTABLE} ${PADDLE_SOURCE_DIR}/tools/test_runner.py ${py_test_modules_MODULES}
                    WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
        else()
            add_test(NAME ${TARGET_NAME}
                    COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python ${py_test_modules_ENVS}
                    ${PYTHON_EXECUTABLE} ${PADDLE_SOURCE_DIR}/tools/test_runner.py ${py_test_modules_MODULES}
                    WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
        endif()
406 407 408
    endif()

    if (py_test_modules_SERIAL)
409
        set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)
410
    endif()
411 412 413
    if(WIN32)
        set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 150)
    endif()
414 415
  endif()
endfunction()
G
gongweibao 已提交
416

417

G
gongweibao 已提交
418 419 420 421 422 423
function(bash_test_modules TARGET_NAME)
    if(NOT WITH_TESTING)
        return()
    endif()

    set(options SERIAL)
424 425
    set(oneValueArgs TIMEOUT START_BASH)
    set(multiValueArgs DEPS ENVS LABELS)
G
gongweibao 已提交
426 427
    cmake_parse_arguments(bash_test_modules "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})

428

429 430 431 432
    set(timeout 350)
    if(${bash_test_modules_TIMEOUT})
        set(timeout ${bash_test_modules_TIMEOUT})
    endif()
G
gongweibao 已提交
433

434 435
    if(WITH_COVERAGE)
        add_test(NAME ${TARGET_NAME}
436
            COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python
437 438
            TEST_TARGET_NAME=${TARGET_NAME} TEST_TIMEOUT=${timeout} ${bash_test_modules_ENVS}
            WITH_COVERAGE=ON COVERAGE_FILE=${PADDLE_BINARY_DIR}/python-coverage.data
439
            bash ${CMAKE_CURRENT_BINARY_DIR}/${bash_test_modules_START_BASH}
440 441 442
            WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
    else()
        add_test(NAME ${TARGET_NAME}
443
            COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python
444
            TEST_TARGET_NAME=${TARGET_NAME} TEST_TIMEOUT=${timeout} ${bash_test_modules_ENVS}
445
            bash ${CMAKE_CURRENT_BINARY_DIR}/${bash_test_modules_START_BASH}
446 447
            WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
    endif()
448

G
gongweibao 已提交
449 450 451
    if (bash_test_modules_SERIAL)
        set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)
    endif()
452 453

    if(bash_test_modules_LABELS)
454
        set_tests_properties(${TARGET_NAME} PROPERTIES LABELS ${bash_test_modules_LABELS})
455
    endif()
G
gongweibao 已提交
456 457
endfunction()

458 459 460 461 462 463 464 465 466 467 468
function(parallel_bash_test_modules TARGET_NAME)
    if(NOT WITH_TESTING)
        return()
    endif()

    set(options SERIAL)
    set(oneValueArgs TIMEOUT START_BASH)
    set(multiValueArgs DEPS ENVS LABELS UnitTests)
    cmake_parse_arguments(parallel_bash_test_modules "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})


Y
YUNSHEN XIE 已提交
469
    set(timeout 120)
470 471 472 473 474 475 476 477
    if(${parallel_bash_test_modules_TIMEOUT})
        set(timeout ${parallel_bash_test_modules_TIMEOUT})
    endif()

    list(JOIN  parallel_bash_test_modules_UnitTests " " uts_string)

    if(WITH_COVERAGE)
        add_test(NAME ${TARGET_NAME}
478
            COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python
479 480 481 482 483 484
            TEST_TARGET_NAME=${TARGET_NAME} TEST_TIMEOUT=${timeout} ${parallel_bash_test_modules_ENVS} UnitTests=${uts_string}
            WITH_COVERAGE=ON COVERAGE_FILE=${PADDLE_BINARY_DIR}/python-coverage.data
            bash ${CMAKE_CURRENT_BINARY_DIR}/${parallel_bash_test_modules_START_BASH}
            WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
    else()
        add_test(NAME ${TARGET_NAME}
485
            COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python
486 487 488 489 490 491 492 493 494 495
            TEST_TARGET_NAME=${TARGET_NAME} TEST_TIMEOUT=${timeout} ${parallel_bash_test_modules_ENVS} UnitTests=${uts_string}
            bash ${CMAKE_CURRENT_BINARY_DIR}/${parallel_bash_test_modules_START_BASH}
            WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
    endif()

    if (parallel_bash_test_modules_SERIAL)
        set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)
    endif()

    if(parallel_bash_test_modules_LABELS)
496
        set_tests_properties(${TARGET_NAME} PROPERTIES LABELS ${parallel_bash_test_modules_LABELS})
497 498 499
    endif()
endfunction()

H
Huihuang Zheng 已提交
500
list(REMOVE_ITEM TEST_OPS test_feed_data_check_shape_type)
501
list(REMOVE_ITEM TEST_OPS test_fetch_lod_tensor_array)
502
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
Y
yuyang18 已提交
503
list(REMOVE_ITEM TEST_OPS test_parallel_executor_crf)
504
list(REMOVE_ITEM TEST_OPS test_parallel_executor_profiler)
H
hutuxian 已提交
505
list(REMOVE_ITEM TEST_OPS test_data_norm_op)
506
list(REMOVE_ITEM TEST_OPS test_parallel_executor_fetch_feed)
507
list(REMOVE_ITEM TEST_OPS test_parallel_executor_transformer)
508
list(REMOVE_ITEM TEST_OPS test_parallel_executor_transformer_auto_growth)
D
dengkaipeng 已提交
509 510
list(REMOVE_ITEM TEST_OPS test_bilinear_interp_op)
list(REMOVE_ITEM TEST_OPS test_nearest_interp_op)
511
list(REMOVE_ITEM TEST_OPS test_imperative_resnet)
512 513
list(REMOVE_ITEM TEST_OPS test_imperative_resnet_sorted_gradient)
list(REMOVE_ITEM TEST_OPS test_imperative_mnist_sorted_gradient)
Y
Yan Xu 已提交
514
list(REMOVE_ITEM TEST_OPS test_imperative_se_resnext)
M
minqiyang 已提交
515
list(REMOVE_ITEM TEST_OPS test_imperative_mnist)
D
dzhwinter 已提交
516
list(REMOVE_ITEM TEST_OPS test_ir_memory_optimize_transformer)
L
lujun 已提交
517
list(REMOVE_ITEM TEST_OPS test_layers)
518 519 520
list(REMOVE_ITEM TEST_OPS test_parallel_executor_seresnext_base_cpu)
list(REMOVE_ITEM TEST_OPS test_parallel_executor_seresnext_with_reduce_cpu)
list(REMOVE_ITEM TEST_OPS test_parallel_executor_seresnext_with_fuse_all_reduce_cpu)
521
list(REMOVE_ITEM TEST_OPS test_imperative_ocr_attention_model)
522
list(REMOVE_ITEM TEST_OPS test_async_ssa_graph_executor_mnist)
523
list(REMOVE_ITEM TEST_OPS test_install_check)
524 525 526 527
list(REMOVE_ITEM TEST_OPS test_basic_gru_api)
list(REMOVE_ITEM TEST_OPS test_basic_gru_unit_op)
list(REMOVE_ITEM TEST_OPS test_basic_lstm_api)
list(REMOVE_ITEM TEST_OPS test_basic_lstm_unit_op)
H
Huihuang Zheng 已提交
528
list(REMOVE_ITEM TEST_OPS test_fuse_all_reduce_pass)
Z
Zhen Wang 已提交
529
list(REMOVE_ITEM TEST_OPS test_fuse_bn_act_pass)
Z
Zhang Ting 已提交
530
list(REMOVE_ITEM TEST_OPS test_fuse_bn_add_act_pass)
531 532
list(REMOVE_ITEM TEST_OPS test_imperative_static_runner_mnist)
list(REMOVE_ITEM TEST_OPS test_imperative_static_runner_while)
Y
YUNSHEN XIE 已提交
533
# disable test_cumsum_op temporaily
534
# list(REMOVE_ITEM TEST_OPS test_cumsum_op)
Z
Zeng Jinle 已提交
535

536 537
# disable this unittest temporarily
list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception)
Y
YUNSHEN XIE 已提交
538

539 540 541 542 543
# disable sparse_attention which not in suitable env
if ( (NOT WITH_GPU) OR (WIN32) OR (PADDLE_WITH_ARM) OR (WITH_ROCM) )
    list(REMOVE_ITEM TEST_OPS test_sparse_attention_op)
endif()

Z
Zeng Jinle 已提交
544 545 546
if (APPLE OR WIN32)
  list(REMOVE_ITEM TEST_OPS test_dataset)
  list(REMOVE_ITEM TEST_OPS test_dataset_dataloader)
547
  list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_base)
548
  # list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception)
549
  list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_process)
550 551
  list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_fds_clear)
  list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exit_func)
552
  list(REMOVE_ITEM TEST_OPS test_imperative_signal_handler)
553 554
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_static)
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_dynamic)
555
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_exception)
556
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_iterable_dataset)
557
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_dataset)
558
  list(REMOVE_ITEM TEST_OPS test_paddle_multiprocessing)
Z
Zeng Jinle 已提交
559 560
endif()

X
xiongkun 已提交
561 562 563 564
if (NOT WITH_GLOO)
    LIST(REMOVE_ITEM TEST_OPS test_cpuonly_spawn)
endif()

565 566 567 568
if(NOT WITH_GPU OR WIN32 OR APPLE)
  list(REMOVE_ITEM TEST_OPS test_build_strategy_fusion_group_pass)
endif()

569
# Some ops need to check results when gc is enabled
570
# Currently, only ops that register NoNeedBufferVarsInference need to do this test
571 572
set(TEST_OPS_WITH_GC
  test_affine_channel_op
573 574 575 576 577
  test_concat_op
  test_elementwise_add_op
  test_elementwise_sub_op
  test_fill_zeros_like2_op
  test_gather_op
578
  test_gather_nd_op
579 580 581 582 583 584 585 586
  test_linear_chain_crf_op
  test_lod_reset_op
  test_lookup_table_op
  test_mean_op
  test_pad2d_op
  test_scatter_op
  test_slice_op
  test_space_to_depth_op
587
  test_squared_l2_distance_op)
588 589 590 591 592 593

foreach(TEST_OP ${TEST_OPS_WITH_GC})
  list(REMOVE_ITEM TEST_OPS ${TEST_OP})
  py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS})
endforeach()

594 595 596 597 598 599 600
# Switch some dy2st UT to eager mode
set(TEST_EAGER_OPS test_jit_save_load test_translated_layer)
foreach(TEST_OP ${TEST_EAGER_OPS})
  list(REMOVE_ITEM TEST_OPS ${TEST_OP})
  py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS FLAGS_enable_eager_mode=1)
endforeach()

601 602 603 604 605 606 607 608 609
if ((NOT WITH_GPU) AND (NOT WITH_XPU) AND NOT (WITH_ASCEND OR WITH_ASCEND_CL))
    list(REMOVE_ITEM TEST_OPS "test_fleet_graph_execution_meta_optimizer")
    list(REMOVE_ITEM TEST_OPS "test_gen_nccl_id_op")
    list(REMOVE_ITEM TEST_OPS "test_dist_fleet_grad_clip")
    list(REMOVE_ITEM TEST_OPS "test_dist_fleet_heter_ctr")
    list(REMOVE_ITEM TEST_OPS "test_dist_fleet_ps_gpu_ctr")
    list(REMOVE_ITEM TEST_OPS "test_dist_mnist_batch_merge")
endif()

Y
yuyang18 已提交
610 611 612
foreach(TEST_OP ${TEST_OPS})
    py_test_modules(${TEST_OP} MODULES ${TEST_OP})
endforeach(TEST_OP)
613
py_test_modules(test_adam_op_multi_thread MODULES test_adam_op ENVS FLAGS_inner_op_parallelism=4)
614
if (WITH_GPU OR WITH_XPU OR WITH_ASCEND OR WITH_ASCEND_CL OR APPLE)
Z
zhangchunle 已提交
615 616 617
    py_test_modules(test_warpctc_op MODULES test_warpctc_op)
    set_tests_properties(test_warpctc_op PROPERTIES TIMEOUT 120)
endif()
618 619
py_test_modules(test_bilinear_interp_op MODULES test_bilinear_interp_op ENVS ${GC_ENVS})
py_test_modules(test_nearest_interp_op MODULES test_nearest_interp_op ENVS ${GC_ENVS})
620
py_test_modules(test_imperative_resnet MODULES test_imperative_resnet ENVS
621
    FLAGS_cudnn_deterministic=1 SERIAL)
622
set_tests_properties(test_imperative_resnet PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
623
py_test_modules(test_imperative_resnet_sorted_gradient MODULES test_imperative_resnet_sorted_gradient ENVS
624
        FLAGS_cudnn_deterministic=1 SERIAL)
625
set_tests_properties(test_imperative_resnet_sorted_gradient PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
M
minqiyang 已提交
626
py_test_modules(test_imperative_mnist MODULES test_imperative_mnist ENVS
627
    FLAGS_cudnn_deterministic=1)
628
py_test_modules(test_imperative_mnist_sorted_gradient MODULES test_imperative_mnist_sorted_gradient ENVS
629
        FLAGS_cudnn_deterministic=1)
Y
Yan Xu 已提交
630
py_test_modules(test_imperative_se_resnext MODULES test_imperative_se_resnext ENVS
631
    FLAGS_cudnn_deterministic=1 SERIAL)
632
set_tests_properties(test_imperative_se_resnext PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
633 634
py_test_modules(test_imperative_ocr_attention_model MODULES test_imperative_ocr_attention_model ENVS
        FLAGS_cudnn_deterministic=1 SERIAL)
635 636 637
py_test_modules(test_install_check MODULES test_install_check ENVS
        FLAGS_cudnn_deterministic=1 SERIAL)
set_tests_properties(test_install_check PROPERTIES LABELS "RUN_TYPE=DIST")
638 639 640 641
py_test_modules(test_imperative_static_runner_mnist MODULES test_imperative_static_runner_mnist ENVS
    FLAGS_cudnn_deterministic=1)
py_test_modules(test_imperative_static_runner_while MODULES test_imperative_static_runner_while ENVS
    FLAGS_cudnn_deterministic=1)
642 643 644 645 646 647 648 649 650

if ((WITH_GPU) AND (CUDA_VERSION GREATER_EQUAL 11.6))
    py_test_modules(test_fused_gemm_epilogue_op MODULES test_fused_gemm_epilogue_op)
    py_test_modules(test_fused_gemm_epilogue_grad_op MODULES test_fused_gemm_epilogue_grad_op)
    py_test_modules(test_fused_gemm_epilogue_op_with_es MODULES test_fused_gemm_epilogue_op ENVS FLAGS_cublaslt_exhaustive_search_times=30)
    py_test_modules(test_fused_gemm_epilogue_grad_op_with_es MODULES test_fused_gemm_epilogue_grad_op ENVS FLAGS_cublaslt_exhaustive_search_times=30)
    py_test_modules(test_fuse_gemm_epilogue_pass MODULES test_fuse_gemm_epilogue_pass)
endif()

651
set_tests_properties(test_conv2d_op PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
652
set_tests_properties(test_faster_tokenizer_op PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
653 654
set_tests_properties(test_conv2d_op_depthwise_conv PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
set_tests_properties(test_conv2d_api PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
C
ceci3 已提交
655 656 657
set_tests_properties(test_conv_nn_grad PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
set_tests_properties(test_norm_nn_grad PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
set_tests_properties(test_nn_grad PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
658
if(WITH_DISTRIBUTE)
659
    add_subdirectory(distributed_passes)
Z
ziyoujiyi 已提交
660
    add_subdirectory(ps)
661 662
    add_subdirectory(auto_parallel)

663 664 665 666
    # FIXME(typhoonzero): add these tests back
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_transformer")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_transpiler")

667 668 669
    # TODO(sandyhouse): fix and add the ut back
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_mnist_hallreduce")

670 671 672 673
    #not need
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_base")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_fleet_base")

T
tangwei12 已提交
674

675 676 677 678 679 680 681 682
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_ctr")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_mnist_lars")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_mnist_train")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_save_load")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_text_classification")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_train")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_word2vec")

T
tangwei12 已提交
683 684
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_fleet_gloo")

T
tangwei12 已提交
685
    py_test_modules(test_recv_save_op MODULES test_recv_save_op ENVS ${dist_ENVS})
686
    py_test_modules(test_communicator_async MODULES test_communicator_async ENVS ${dist_ENVS})
T
Thunderbrook 已提交
687
    py_test_modules(test_communicator_ps_gpu MODULES test_communicator_ps_gpu ENVS ${dist_ENVS})
688 689
    py_test_modules(test_communicator_geo MODULES test_communicator_geo ENVS ${dist_ENVS})
    py_test_modules(test_communicator_half_async MODULES test_communicator_half_async ENVS ${dist_ENVS} FLAGS_communicator_send_queue_size=1 FLAGS_communicator_max_merge_var_num=1)
T
tangwei12 已提交
690
    py_test_modules(test_communicator_sync MODULES test_communicator_sync ENVS ${dist_ENVS} FLAGS_communicator_send_queue_size=1 FLAGS_communicator_max_merge_var_num=1)
M
mapingshuo 已提交
691
    py_test_modules(test_collective_optimizer MODULES test_collective_optimizer)
692
    if(NOT APPLE)
693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710
        py_test_modules(test_fleet_base MODULES test_fleet_base ENVS ${dist_ENVS})
        py_test_modules(test_fleet_base_2 MODULES test_fleet_base_2 ENVS ${dist_ENVS})
        py_test_modules(test_fleet_base_3 MODULES test_fleet_base_3 ENVS ${dist_ENVS})
        py_test_modules(test_fleet_amp_init MODULES test_fleet_amp_init ENVS ${dist_ENVS})
        py_test_modules(test_fleet_fp16_allreduce_meta_optimizer MODULES test_fleet_fp16_allreduce_meta_optimizer ENVS ${dist_ENVS})
        py_test_modules(test_fleet_private_function MODULES test_fleet_private_function ENVS ${dist_ENVS})
        py_test_modules(test_fleet_meta_optimizer_base MODULES test_fleet_meta_optimizer_base ENVS ${dist_ENVS})
        py_test_modules(test_fleet_distributed_strategy MODULES test_fleet_distributed_strategy)
        py_test_modules(test_fleet_static_mp_layers MODULES test_fleet_static_mp_layers)
        #py_test_modules(test_fleet_auto MODULES test_fleet_auto ENVS ${dist_ENVS})
        if (WITH_GPU OR WITH_XPU OR WITH_ASCEND OR WITH_ASCEND_CL)
            py_test_modules(test_fleet_amp_meta_optimizer MODULES test_fleet_amp_meta_optimizer ENVS ${dist_ENVS})
            py_test_modules(test_fleet_gradient_merge_meta_optimizer MODULES test_fleet_gradient_merge_meta_optimizer ENVS ${dist_ENVS})
            py_test_modules(test_fleet_graph_executor MODULES test_fleet_graph_executor ENVS ${dist_ENVS})
            py_test_modules(test_fleet_hybrid_meta_optimizer MODULES test_fleet_hybrid_meta_optimizer ENVS ${dist_ENVS})
            py_test_modules(test_fleet_recompute_meta_optimizer MODULES test_fleet_recompute_meta_optimizer ENVS ${dist_ENVS})
            py_test_modules(test_fleet_sharding_meta_optimizer MODULES test_fleet_sharding_meta_optimizer ENVS ${dist_ENVS})
        endif()
Y
Yi Liu 已提交
711
        if(NOT WIN32)
712 713
            py_test_modules(test_auto_parallel_partitioner MODULES test_auto_parallel_partitioner ENVS ${dist_ENVS})
            py_test_modules(test_auto_parallel_partitioner_gpt MODULES test_auto_parallel_partitioner_gpt ENVS ${dist_ENVS})
C
caozhou 已提交
714
            py_test_modules(test_auto_parallel_searcher MODULES test_auto_parallel_searcher ENVS ${dist_ENVS})
C
caozhou 已提交
715
            py_test_modules(test_auto_parallel_reshard MODULES test_auto_parallel_reshard ENVS ${dist_ENVS})
716
            py_test_modules(test_auto_parallel_dist_tensor MODULES test_auto_parallel_dist_tensor ENVS ${dist_ENVS})
C
caozhou 已提交
717 718 719
            py_test_modules(test_auto_parallel_reshard_serial MODULES test_auto_parallel_reshard_serial ENVS ${dist_ENVS})
            py_test_modules(test_auto_parallel_reshard_mppp MODULES test_auto_parallel_reshard_mppp ENVS ${dist_ENVS})
            py_test_modules(test_auto_parallel_reshard_dpmppp MODULES test_auto_parallel_reshard_dpmppp ENVS ${dist_ENVS})
720
            py_test_modules(test_auto_parallel_cost_model MODULES test_auto_parallel_cost_model ENVS ${dist_ENVS})
721 722 723 724 725 726 727 728
            if (WITH_GPU OR WITH_XPU OR WITH_ASCEND OR WITH_ASCEND_CL)
                py_test_modules(test_fleet_lamb_meta_optimizer MODULES test_fleet_lamb_meta_optimizer ENVS ${dist_ENVS})
                py_test_modules(test_fleet_lars_meta_optimizer MODULES test_fleet_lars_meta_optimizer ENVS ${dist_ENVS})
                py_test_modules(test_fleet_localsgd_meta_optimizer MODULES test_fleet_localsgd_meta_optimizer ENVS ${dist_ENVS})



            endif()
Y
Yi Liu 已提交
729
        endif(NOT WIN32)
730
    endif(NOT APPLE)
G
gongweibao 已提交
731
    if(WITH_DGC)
W
WangXi 已提交
732 733
        # if with dgc, test all dgc tests.
        # NOTE. dist dgc tests is already in DIST_TEST_OPS
G
gongweibao 已提交
734
        py_test_modules(test_dgc_op MODULES test_dgc_op)
735 736
        py_test_modules(test_dgc_momentum_op MODULES test_dgc_momentum_op)
        py_test_modules(test_dgc_optimizer MODULES test_dgc_optimizer)
737
        py_test_modules(test_fleet_dgc_meta_optimizer MODULES test_fleet_dgc_meta_optimizer)
W
WangXi 已提交
738 739 740 741
    else()
        # if not with dgc, must close all dgc tests
        list(REMOVE_ITEM DIST_TEST_OPS "test_dist_mnist_dgc_nccl")
        list(REMOVE_ITEM DIST_TEST_OPS "test_dist_se_resnext_dgc")
G
gongweibao 已提交
742
    endif()
J
JiabinYang 已提交
743
    if(NOT APPLE)
744
        if(WITH_GPU OR WITH_ROCM)
745
            bash_test_modules(test_c_comm_init_op START_BASH test_c_comm_init_op.sh  ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
746
            py_test_modules(test_launch_coverage MODULES test_launch_coverage)
W
WangXi 已提交
747
        endif()
748

749
        bash_test_modules(test_fleetrun START_BASH test_fleetrun.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
750 751
        if (WITH_GPU OR WITH_XPU OR WITH_ASCEND OR WITH_ASCEND_CL)
            bash_test_modules(test_fleet_launch_nproc START_BASH test_fleet_launch_nproc.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
752 753 754
            bash_test_modules(test_fleet_run_random_port START_BASH test_fleet_run_random_port.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
            bash_test_modules(test_fleet_launch_async START_BASH test_fleet_launch_async.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
            bash_test_modules(test_fleet_launch_cloud START_BASH test_fleet_launch_cloud.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
755
        endif()
756
        if(WITH_ASCEND OR WITH_ASCEND_CL)
757 758 759
            bash_test_modules(test_fleet_launch_ascend START_BASH test_fleet_launch_ascend.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
            bash_test_modules(test_ascend_group START_BASH test_ascend_group.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
        endif()
760

761
        # port range (20000, 23000) is reserved for dist-ops
G
gongweibao 已提交
762
        set(dist_ut_port 20001)
763
        foreach(TEST_OP ${DIST_TEST_OPS})
764
            bash_test_modules(${TEST_OP} START_BASH dist_test.sh SERIAL LABELS "RUN_TYPE=EXCLUSIVE" ENVS "PADDLE_DIST_UT_PORT=${dist_ut_port}")
765
            MATH(EXPR dist_ut_port "${dist_ut_port}+20")
766 767 768
            if(dist_ut_port GREATER_EQUAL 22998)
                message(FATAL_ERROR "available ports have been exhausted:${dist_ut_port}")
            endif()
769
        endforeach(TEST_OP)
770
        # solve it later.
771
        bash_test_modules(test_fleet_launch_ps START_BASH test_fleet_launch_ps.sh SERIAL LABELS "RUN_TYPE=EXCLUSIVE" ENVS "PADDLE_DIST_UT_PORT=${dist_ut_port}" PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR} )
X
xiongkun 已提交
772 773 774
        if (WITH_GLOO)
            bash_test_modules(test_cpuonly_launch START_BASH test_cpuonly_launch.sh SERIAL LABELS "RUN_TYPE=EXCLUSIVE" ENVS "PADDLE_DIST_UT_PORT=${dist_ut_port}" PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR} )
        endif()
775 776 777
        if (WITH_GPU OR WITH_XPU OR WITH_ASCEND OR WITH_ASCEND_CL)
            bash_test_modules(test_new_group START_BASH test_new_group.sh SERIAL LABELS "RUN_TYPE=EXCLUSIVE" ENVS "PADDLE_DIST_UT_PORT=${dist_ut_port}+20" PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR} )
        endif()
778
    endif(NOT APPLE)
779
endif()
780

781
py_test_modules(test_parallel_executor_crf MODULES test_parallel_executor_crf)
782 783
# Coverage pipeline use cuda 10.1 now, profiler will random hang in cuda 10.1,
# see https://github.com/PaddlePaddle/Paddle/issues/29082 for details.
784
# We guess there are some bugs in cuda 10.1 or 10.2,
785 786 787 788 789 790
# since this unittest is stable in cuda 11 (py3 pipeline) now.
if(NOT WITH_COVERAGE)
  py_test_modules(test_parallel_executor_profiler MODULES test_parallel_executor_profiler)
  set_tests_properties(test_parallel_executor_profiler PROPERTIES LABELS "RUN_TYPE=DIST")
  set_tests_properties(test_parallel_executor_profiler PROPERTIES TIMEOUT 120)
endif()
791
py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer)
792 793
if(WIN32)
    py_test_modules(test_parallel_executor_transformer_auto_growth MODULES test_parallel_executor_transformer_auto_growth ENVS FLAGS_allocator_strategy=auto_growth CUDA_VISIBLE_DEVICES=0)
H
Huihuang Zheng 已提交
794 795
    py_test_modules(test_fuse_all_reduce_pass MODULES test_fuse_all_reduce_pass ENVS CUDA_VISIBLE_DEVICES=0)
    py_test_modules(test_feed_data_check_shape_type MODULES test_feed_data_check_shape_type ENVS CUDA_VISIBLE_DEVICES=0)
796
    py_test_modules(test_fetch_lod_tensor_array MODULES test_fetch_lod_tensor_array ENVS CUDA_VISIBLE_DEVICES=0)
797 798
else()
    py_test_modules(test_parallel_executor_transformer_auto_growth MODULES test_parallel_executor_transformer_auto_growth ENVS FLAGS_allocator_strategy=auto_growth)
H
Huihuang Zheng 已提交
799 800
    py_test_modules(test_fuse_all_reduce_pass MODULES test_fuse_all_reduce_pass)
    py_test_modules(test_feed_data_check_shape_type MODULES test_feed_data_check_shape_type)
801
    py_test_modules(test_fetch_lod_tensor_array MODULES test_fetch_lod_tensor_array)
802
endif()
803

H
hutuxian 已提交
804
py_test_modules(test_data_norm_op MODULES test_data_norm_op)
Z
Zhen Wang 已提交
805
py_test_modules(test_fuse_bn_act_pass MODULES test_fuse_bn_act_pass ENVS FLAGS_cudnn_deterministic=1 FLAGS_cudnn_batchnorm_spatial_persistent=1 FLAGS_conv_workspace_size_limit=1000)
Z
Zhang Ting 已提交
806
py_test_modules(test_fuse_bn_add_act_pass MODULES test_fuse_bn_add_act_pass ENVS FLAGS_cudnn_deterministic=1 FLAGS_cudnn_batchnorm_spatial_persistent=1 FLAGS_conv_workspace_size_limit=1000)
807

808
# NOTE: These unittests will appear NaN steadily in windows CI. After analysis,
809 810
# it is found that windows CI will run all the training unittests with the ON_INFER option turned on,
# which will not appear in other CIs. The calculation behavior of some ops in inference mode is
811 812
# inconsistent with that in non-inference mode.
if(NOT ON_INFER)
813 814 815 816
    py_test_modules(test_parallel_executor_seresnext_base_cpu MODULES test_parallel_executor_seresnext_base_cpu)
    py_test_modules(test_parallel_executor_seresnext_with_reduce_cpu MODULES test_parallel_executor_seresnext_with_reduce_cpu)
    py_test_modules(test_parallel_executor_seresnext_with_fuse_all_reduce_cpu MODULES test_parallel_executor_seresnext_with_fuse_all_reduce_cpu)
    set_tests_properties(test_parallel_executor_seresnext_base_cpu PROPERTIES TIMEOUT 900)
817
    set_tests_properties(test_parallel_executor_seresnext_base_cpu PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
818
    set_tests_properties(test_parallel_executor_seresnext_with_reduce_cpu PROPERTIES TIMEOUT 750)
819
    set_tests_properties(test_parallel_executor_seresnext_with_reduce_cpu PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
820
    set_tests_properties(test_parallel_executor_seresnext_with_fuse_all_reduce_cpu PROPERTIES TIMEOUT 750)
821
    set_tests_properties(test_parallel_executor_seresnext_with_fuse_all_reduce_cpu PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
822
endif()
823

824 825 826
if(NOT WIN32)
    # TODO: fix these unittests failure on Windows
    py_test_modules(test_layers MODULES test_layers ENVS FLAGS_cudnn_deterministic=1)
827
    py_test_modules(test_ir_memory_optimize_transformer MODULES test_ir_memory_optimize_transformer)
828 829 830
    # FIXME(zcd): temporally disable test_parallel_executor_fetch_feed in Windows CI because of the random failure.
    py_test_modules(test_parallel_executor_fetch_feed MODULES test_parallel_executor_fetch_feed)
    set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 450)
D
dzhwinter 已提交
831
endif()
832

833
if(WITH_DISTRIBUTE AND NOT APPLE AND NOT WIN32)
G
gongweibao 已提交
834 835 836 837 838 839 840 841 842 843 844 845
    py_test_modules(test_fleet_checkpoint MODULES test_fleet_checkpoint)
    set_tests_properties(test_fleet_checkpoint PROPERTIES TIMEOUT 200)
    set_tests_properties(test_fleet_checkpoint PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_auto_checkpoint START_BASH dist_test.sh TIMEOUT 200 LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_auto_checkpoint1 START_BASH dist_test.sh TIMEOUT 200  LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_auto_checkpoint2 START_BASH dist_test.sh TIMEOUT 200  LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_auto_checkpoint3 START_BASH dist_test.sh TIMEOUT 200  LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_auto_checkpoint_multiple START_BASH dist_test.sh TIMEOUT 200  LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_auto_checkpoint_dist_basic START_BASH dist_test.sh TIMEOUT 200  LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_hdfs1 START_BASH dist_test.sh TIMEOUT 200  LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_hdfs2 START_BASH dist_test.sh TIMEOUT 200   LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
    bash_test_modules(test_hdfs3 START_BASH dist_test.sh TIMEOUT 200  LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
846 847
endif()

848
add_subdirectory(sequence)
849
add_subdirectory(dygraph_to_static)
F
Feiyu Chan 已提交
850
add_subdirectory(rnn)
851
add_subdirectory(autograd)
852
add_subdirectory(distribution)
853 854 855 856

if (NOT WIN32 OR NOT WITH_GPU)
    add_subdirectory(fft)
endif()
857

T
tianshuo78520a 已提交
858 859 860 861
if (WITH_XPU)
    add_subdirectory(xpu)
endif()

862
# dist xpu tests:
863
if (WITH_XPU_BKCL)
864
    #py_test(test_collective_reduce_api_xpu SRCS "test_collective_reduce_api.py")
865 866 867
    py_test(test_collective_allreduce_api_xpu SRCS "test_collective_allreduce_api.py")
endif()

868 869 870 871
if(WIN32)
    cc_test(cc_imp_py_test SRCS cc_imp_py_test.cc DEPS python)
endif()

872 873 874 875
if (WITH_ASCEND_CL)
    add_subdirectory(npu)
endif()

876 877 878
if (WITH_MKLDNN)
    add_subdirectory(mkldnn)
endif()
879

J
jianghaicheng 已提交
880 881 882 883
if (WITH_IPU)
    add_subdirectory(ipu)
endif()

884 885 886 887
if (WITH_MLU)
    add_subdirectory(mlu)
endif()

888 889
add_subdirectory(asp)

890 891
add_subdirectory(ir)

H
hong 已提交
892 893
add_subdirectory(interpreter)

894
if (WITH_TESTING)
Z
Zeng Jinle 已提交
895
    set_property(TEST test_parallel_executor_mnist PROPERTY ENVIRONMENT GLOG_vmodule=all_reduce_deps_pass=10)
Z
Zeng Jinle 已提交
896
    set_property(TEST test_parallel_executor_fix_op_run_order PROPERTY ENVIRONMENT GLOG_vmodule=fix_op_run_order_pass=10)
897 898
endif()

899
set_tests_properties(test_parallel_executor_test_while_train test_parallel_executor_mnist
900
        test_parallel_executor_feed_persistable_var
901
        test_buffer_shared_memory_reuse_pass_and_fuse_optimization_op_pass
902
        test_data_norm_op
903
        test_dataloader_keep_order
904 905
        test_dataloader_unkeep_order
        test_parallel_executor_inference_feed_partial_data
906
        test_parallel_ssa_graph_inference_feed_partial_data
907
        test_fetch_unmerged
908
        test_buffer_shared_memory_reuse_pass PROPERTIES LABELS "RUN_TYPE=DIST")
909 910
# disable test_parallel_executor_fetch_isolated_var
# set_tests_properties(test_parallel_executor_fetch_isolated_var PROPERTIES LABELS "RUN_TYPE=DIST")
K
Kaipeng Deng 已提交
911
set_tests_properties(test_parallel_executor_crf test_sync_batch_norm_op test_inplace_abn_op
912 913 914
        test_parallel_executor_seresnext_base_gpu
        test_parallel_executor_seresnext_with_reduce_gpu
        test_parallel_executor_seresnext_with_fuse_all_reduce_gpu
915 916
        test_distributed_fused_lamb_op_with_clip
        test_distributed_fused_lamb_op_without_clip
917
        test_distributed_fused_lamb_op_with_gradient_merge
918
        test_parallel_executor_fetch_isolated_var
C
chalsliu 已提交
919
        PROPERTIES LABELS "RUN_TYPE=DIST")
920

921
if(NOT WIN32 AND NOT APPLE)
922
    set_tests_properties(test_imperative_signal_handler PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
C
chalsliu 已提交
923 924 925 926 927 928
    set_tests_properties(test_imperative_data_loader_base PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
    set_tests_properties(test_imperative_data_loader_fds_clear PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
    # set_tests_properties(test_imperative_data_loader_exception PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
    set_tests_properties(test_multiprocess_dataloader_static PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
    set_tests_properties(test_multiprocess_dataloader_dynamic PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
    set_tests_properties(test_multiprocess_dataloader_exception PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
929 930
    set_tests_properties(test_multiprocess_dataloader_iterable_dataset_static PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
    set_tests_properties(test_multiprocess_dataloader_iterable_dataset_dynamic PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
931
    set_tests_properties(test_multiprocess_dataloader_dataset PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
932
    set_tests_properties(test_multiprocess_dataloader_static PROPERTIES TIMEOUT 120)
933
endif()
Y
YUNSHEN XIE 已提交
934

935
if (NOT WIN32)
936
    set_tests_properties(test_multiprocess_reader_exception PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
937
    set_tests_properties(test_layers PROPERTIES TIMEOUT 120)
F
feng_shuai 已提交
938 939 940
    if (WITH_NV_JETSON)
    set_tests_properties(test_ir_memory_optimize_transformer PROPERTIES TIMEOUT 1200)
    else ()
941
    set_tests_properties(test_ir_memory_optimize_transformer PROPERTIES TIMEOUT 120)
F
feng_shuai 已提交
942
    endif ()
943 944 945
endif()

if (WITH_DISTRIBUTE AND NOT WIN32)
946
    set_tests_properties(test_fleet_utils PROPERTIES TIMEOUT 120)
947
    set_tests_properties(test_collective_cpu_barrier_with_gloo PROPERTIES TIMEOUT 40)
948 949 950 951
endif()

if (WITH_DISTRIBUTE)
    set_tests_properties(test_communicator_half_async PROPERTIES TIMEOUT 120)
952 953
    set_tests_properties(test_dist_fleet_ctr2 PROPERTIES TIMEOUT 200)
    set_tests_properties(test_dist_fleet_sparse_embedding_ctr PROPERTIES TIMEOUT 200)
954
    set_tests_properties(test_dist_fleet_infer PROPERTIES TIMEOUT 200)
955
    set_tests_properties(test_dist_fleet_raw_program_optimizer PROPERTIES TIMEOUT 120)
956
    set_tests_properties(test_dist_fleet_raw_program_optimizer_fuse_allreduce PROPERTIES TIMEOUT 60)
957
    set_tests_properties(test_dist_dygraph_apis PROPERTIES TIMEOUT 120)
958
endif()
959

960
if (WITH_DISTRIBUTE AND NOT APPLE)
961
    if(WITH_GPU OR WITH_ROCM)
962
        set_tests_properties(test_c_comm_init_op PROPERTIES TIMEOUT 120)
963
        set_tests_properties(test_dist_mnist_gradient_merge PROPERTIES TIMEOUT 360)
964
    endif()
Y
YUNSHEN XIE 已提交
965 966 967
endif()

# setting timeout value as 15S
968
set_tests_properties(test_run PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
969 970
set_tests_properties(test_sync_batch_norm_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_cross_op PROPERTIES TIMEOUT 120)
971
set_tests_properties(test_imperative_lod_tensor_to_selected_rows PROPERTIES TIMEOUT 200)
Y
YUNSHEN XIE 已提交
972 973
set_tests_properties(test_lstm_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_star_gan_with_gradient_penalty PROPERTIES TIMEOUT 120)
Z
zhangchunle 已提交
974

Y
YUNSHEN XIE 已提交
975
set_tests_properties(test_bicubic_interp_op PROPERTIES TIMEOUT 120)
976
set_tests_properties(test_deformable_conv_op PROPERTIES TIMEOUT 200)
Y
YUNSHEN XIE 已提交
977 978 979 980
set_tests_properties(test_nearest_interp_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_profiler PROPERTIES TIMEOUT 120)
set_tests_properties(test_inplace_softmax_with_cross_entropy PROPERTIES TIMEOUT 120)
set_tests_properties(test_cross_entropy2_op PROPERTIES TIMEOUT 120)
981
set_tests_properties(test_cross_entropy_loss PROPERTIES TIMEOUT 180)
Y
YUNSHEN XIE 已提交
982 983
set_tests_properties(test_fetch_unmerged PROPERTIES TIMEOUT 120)
set_tests_properties(test_gru_unit_op PROPERTIES TIMEOUT 120)
D
Double_V 已提交
984
set_tests_properties(test_activation_nn_grad PROPERTIES TIMEOUT 200)
Y
YUNSHEN XIE 已提交
985 986 987
set_tests_properties(test_empty_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_executor_transformer PROPERTIES TIMEOUT 120)
set_tests_properties(test_elementwise_div_op PROPERTIES TIMEOUT 120)
988
set_tests_properties(test_regularizer_api PROPERTIES TIMEOUT 150)
Y
YUNSHEN XIE 已提交
989
set_tests_properties(test_multiclass_nms_op PROPERTIES TIMEOUT 120)
990
if(NOT WIN32)
F
feng_shuai 已提交
991 992 993
    if (WITH_NV_JETSON)
    set_tests_properties(test_ir_memory_optimize_nlp PROPERTIES TIMEOUT 1200)
    else ()
994
    set_tests_properties(test_ir_memory_optimize_nlp PROPERTIES TIMEOUT 120)
F
feng_shuai 已提交
995
    endif ()
996
endif()
Y
YUNSHEN XIE 已提交
997 998 999 1000 1001 1002 1003 1004 1005 1006 1007
set_tests_properties(test_add_reader_dependency PROPERTIES TIMEOUT 120)
set_tests_properties(test_bilateral_slice_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_buffer_shared_memory_reuse_pass PROPERTIES TIMEOUT 120)
set_tests_properties(test_fuse_relu_depthwise_conv_pass PROPERTIES TIMEOUT 120)
set_tests_properties(test_fleet_util PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_transformer_sorted_gradient PROPERTIES TIMEOUT 120)
set_tests_properties(test_matmul_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_nearest_interp_v2_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_trilinear_interp_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_bicubic_interp_v2_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_gather_op PROPERTIES TIMEOUT 120)
1008
set_tests_properties(test_static_save_load PROPERTIES TIMEOUT 250)
1009
set_tests_properties(test_pylayer_op PROPERTIES TIMEOUT 120)
1010
set_tests_properties(test_paddle_save_load_binary PROPERTIES TIMEOUT 120)
W
WeiXin 已提交
1011
if (WIN32)
W
WeiXin 已提交
1012
    set_tests_properties(test_static_save_load_large PROPERTIES TIMEOUT 900)
W
WeiXin 已提交
1013 1014
    set_tests_properties(test_paddle_save_load PROPERTIES TIMEOUT 250)
else()
W
WeiXin 已提交
1015
    set_tests_properties(test_static_save_load_large PROPERTIES TIMEOUT 600)
1016
    set_tests_properties(test_paddle_save_load PROPERTIES TIMEOUT 250)
W
WeiXin 已提交
1017
endif()
F
feng_shuai 已提交
1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034
if (WITH_NV_JETSON)
    set_tests_properties(test_concat_op PROPERTIES TIMEOUT 1200)
    set_tests_properties(test_conv3d_transpose_part2_op PROPERTIES TIMEOUT 1200)
    set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 1200)
    set_tests_properties(test_conv3d_op PROPERTIES TIMEOUT 1200)
    set_tests_properties(test_norm_op PROPERTIES TIMEOUT 1200)
    set_tests_properties(test_layer_norm_op PROPERTIES TIMEOUT 1500)
    set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 1500)
else()
    set_tests_properties(test_concat_op PROPERTIES TIMEOUT 120)
    set_tests_properties(test_conv3d_transpose_part2_op PROPERTIES TIMEOUT 120)
    set_tests_properties(test_conv3d_transpose_op PROPERTIES TIMEOUT 120)
    set_tests_properties(test_conv3d_op PROPERTIES TIMEOUT 120)
    set_tests_properties(test_norm_op PROPERTIES TIMEOUT 120)
    set_tests_properties(test_layer_norm_op PROPERTIES TIMEOUT 150)
    set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 150)
endif()
1035
set_tests_properties(test_imperative_selected_rows_to_lod_tensor PROPERTIES TIMEOUT 200)
Y
YUNSHEN XIE 已提交
1036 1037 1038 1039 1040 1041 1042 1043 1044
set_tests_properties(test_index_select_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_ssa_graph_inference_feed_partial_data PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_executor_crf PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_save_load PROPERTIES TIMEOUT 120)
set_tests_properties(test_partial_eager_deletion_transformer PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_executor_seresnext_with_reduce_gpu PROPERTIES TIMEOUT 120)
set_tests_properties(test_dropout_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_argsort_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_gather_nd_op PROPERTIES TIMEOUT 120)
1045
set_tests_properties(test_nn_grad PROPERTIES TIMEOUT 180)
Y
YUNSHEN XIE 已提交
1046 1047 1048
set_tests_properties(test_elementwise_sub_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_row_conv_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_executor_seresnext_with_fuse_all_reduce_gpu PROPERTIES TIMEOUT 120)
1049 1050
set_tests_properties(test_distributed_fused_lamb_op_with_clip PROPERTIES TIMEOUT 120)
set_tests_properties(test_distributed_fused_lamb_op_without_clip PROPERTIES TIMEOUT 120)
1051
set_tests_properties(test_distributed_fused_lamb_op_with_gradient_merge PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
1052 1053
set_tests_properties(test_elementwise_min_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_nan_inf PROPERTIES TIMEOUT 120)
1054
set_tests_properties(test_deformable_conv_v1_op PROPERTIES TIMEOUT 300)
Y
YUNSHEN XIE 已提交
1055 1056 1057 1058 1059 1060 1061 1062 1063 1064
set_tests_properties(test_parallel_executor_transformer_auto_growth PROPERTIES TIMEOUT 120)
set_tests_properties(test_py_reader_using_executor PROPERTIES TIMEOUT 120)
set_tests_properties(test_elementwise_add_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_weight_decay PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_ptb_rnn_sorted_gradient PROPERTIES TIMEOUT 120)
set_tests_properties(test_crop_tensor_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_eager_deletion_lstm_net PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_executor_mnist PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_ptb_rnn PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_save_load_v2 PROPERTIES TIMEOUT 120)
K
Kaipeng Deng 已提交
1065
set_tests_properties(test_conv2d_transpose_op PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
1066 1067 1068 1069 1070
set_tests_properties(test_prroi_pool_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_multiprocess_dataloader_iterable_dataset_static PROPERTIES TIMEOUT 120)
set_tests_properties(test_lstm_cudnn_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_stack_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_bilinear_interp_v2_op PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
1071
set_tests_properties(test_svd_op PROPERTIES TIMEOUT 80)
1072
set_tests_properties(test_einsum_op PROPERTIES TIMEOUT 120)
1073
set_tests_properties(test_qr_op PROPERTIES TIMEOUT 60)
Y
YUNSHEN XIE 已提交
1074 1075 1076 1077 1078
set_tests_properties(test_deformable_psroi_pooling PROPERTIES TIMEOUT 120)
set_tests_properties(test_trilinear_interp_v2_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_static_runner_mnist PROPERTIES TIMEOUT 120)
set_tests_properties(test_masked_select_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_sigmoid_cross_entropy_with_logits_op PROPERTIES TIMEOUT 120)
1079
set_tests_properties(test_imperative_optimizer_v2 PROPERTIES TIMEOUT 150)
Y
YUNSHEN XIE 已提交
1080 1081 1082 1083 1084 1085
set_tests_properties(test_partial_sum_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_cond PROPERTIES TIMEOUT 120)
set_tests_properties(test_space_to_depth_op PROPERTIES TIMEOUT 200)
set_tests_properties(test_dyn_rnn PROPERTIES TIMEOUT 120)
set_tests_properties(test_sgd_op PROPERTIES TIMEOUT 250)
set_tests_properties(test_parallel_executor_seresnext_base_gpu PROPERTIES TIMEOUT 120)
1086
set_tests_properties(test_norm_nn_grad PROPERTIES TIMEOUT 180)
Y
YUNSHEN XIE 已提交
1087 1088 1089 1090
set_tests_properties(test_matrix_nms_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_generator_dataloader PROPERTIES TIMEOUT 120)
set_tests_properties(test_partial_concat_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_fuse_optimizer_pass PROPERTIES TIMEOUT 120)
1091
set_tests_properties(test_softmax_with_cross_entropy_op PROPERTIES TIMEOUT 220)
F
From00 已提交
1092
set_tests_properties(test_reduce_op PROPERTIES TIMEOUT 500)
Y
YUNSHEN XIE 已提交
1093 1094 1095 1096 1097 1098
set_tests_properties(test_adam_optimizer_fp32_fp64 PROPERTIES TIMEOUT 120)
set_tests_properties(test_elementwise_nn_grad PROPERTIES TIMEOUT 120)
set_tests_properties(test_buffer_shared_memory_reuse_pass_and_fuse_optimization_op_pass PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv_nn_grad PROPERTIES TIMEOUT 120)
set_tests_properties(test_program_prune_backward PROPERTIES TIMEOUT 120)
set_tests_properties(test_group_norm_op PROPERTIES TIMEOUT 120)
J
Jiabin Yang 已提交
1099 1100
set_tests_properties(test_imperative_optimizer PROPERTIES TIMEOUT 250)
set_tests_properties(test_imperative_optimizer_v2 PROPERTIES TIMEOUT 250)
Y
YUNSHEN XIE 已提交
1101 1102 1103
set_tests_properties(test_pool2d_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_transpose_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_eager_deletion_gru_net PROPERTIES TIMEOUT 120)
S
Shang Zhizhou 已提交
1104
set_tests_properties(test_activation_op PROPERTIES TIMEOUT 270)
Y
YUNSHEN XIE 已提交
1105 1106 1107 1108 1109 1110
set_tests_properties(test_normal PROPERTIES TIMEOUT 120)
set_tests_properties(test_lstmp_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_bilinear_interp_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_decoupled_py_reader PROPERTIES TIMEOUT 120)
set_tests_properties(test_fuse_bn_act_pass PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv2d_op PROPERTIES TIMEOUT 120)
1111 1112
set_tests_properties(test_conv2d_op_depthwise_conv PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv2d_api PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
1113 1114 1115 1116 1117 1118
set_tests_properties(test_elementwise_mul_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_cyclic_cifar_dataset PROPERTIES TIMEOUT 120)
set_tests_properties(test_fuse_all_reduce_pass PROPERTIES TIMEOUT 120)
set_tests_properties(test_dygraph_multi_forward PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_ocr_attention_model PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_mnist PROPERTIES TIMEOUT 120)
1119
set_tests_properties(test_fused_elemwise_activation_op PROPERTIES TIMEOUT 270)
1120
set_tests_properties(test_fused_elemwise_activation_op PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
1121 1122
set_tests_properties(test_gru_op PROPERTIES TIMEOUT 200)
set_tests_properties(test_regularizer PROPERTIES TIMEOUT 150)
Y
YUNSHEN XIE 已提交
1123 1124 1125 1126 1127
set_tests_properties(test_imperative_resnet PROPERTIES TIMEOUT 200)
set_tests_properties(test_imperative_resnet_sorted_gradient PROPERTIES TIMEOUT 200)
set_tests_properties(test_imperative_se_resnext PROPERTIES TIMEOUT 200)
set_tests_properties(test_matmul_v2_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_slice_op PROPERTIES TIMEOUT 120)
1128 1129 1130 1131 1132 1133 1134
set_tests_properties(test_strided_slice_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_translated_layer PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_executor_inference_feed_partial_data PROPERTIES TIMEOUT 120)
set_tests_properties(test_pad3d_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_dataloader_keep_order PROPERTIES TIMEOUT 120)
set_tests_properties(test_mean_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_dataloader_unkeep_order PROPERTIES TIMEOUT 120)
1135 1136
set_tests_properties(test_reader_reset PROPERTIES TIMEOUT 120)
set_tests_properties(test_pool3d_api PROPERTIES TIMEOUT 120)
H
hlygit66666 已提交
1137
set_tests_properties(test_cumprod_op PROPERTIES TIMEOUT 120)
1138
set_tests_properties(test_split_program PROPERTIES TIMEOUT 120)
1139
if(WITH_DISTRIBUTE AND WITH_GPU AND WITH_NCCL)
1140
    set_tests_properties(test_parallel_dygraph_dataparallel PROPERTIES TIMEOUT 120)
1141 1142 1143 1144 1145
    set_tests_properties(test_parallel_dygraph_mnist PROPERTIES TIMEOUT 200)
    set_tests_properties(test_parallel_dygraph_se_resnext PROPERTIES TIMEOUT 200)
    set_tests_properties(test_parallel_dygraph_unused_variables PROPERTIES TIMEOUT 350)
    set_tests_properties(test_parallel_dygraph_control_flow PROPERTIES TIMEOUT 350)
    set_tests_properties(test_parallel_dygraph_no_sync PROPERTIES TIMEOUT 300)
1146
    set_tests_properties(test_parallel_dygraph_no_sync_gradient_check PROPERTIES TIMEOUT 30)
S
ShenLiang 已提交
1147
    set_tests_properties(test_parallel_dygraph_pipeline_parallel PROPERTIES TIMEOUT 500)
1148
    set_tests_properties(test_parallel_dygraph_tensor_parallel PROPERTIES TIMEOUT 200)
J
JZ-LIANG 已提交
1149
    set_tests_properties(test_parallel_dygraph_sharding_parallel PROPERTIES TIMEOUT 120)
1150
    set_tests_properties(test_dygraph_sharding_optimizer_stage2 PROPERTIES TIMEOUT 120)
B
Baibaifan 已提交
1151
    set_tests_properties(test_dygraph_sharding_stage2 PROPERTIES TIMEOUT 120)
B
Baibaifan 已提交
1152
    set_tests_properties(test_dygraph_sharding_stage3 PROPERTIES TIMEOUT 200)
B
Baibaifan 已提交
1153
    set_tests_properties(test_dygraph_group_sharded_api PROPERTIES TIMEOUT 120)
1154
    set_tests_properties(test_auto_parallel_parallelizer PROPERTIES TIMEOUT 120)
1155
    set_tests_properties(test_parallel_dygraph_mp_layers PROPERTIES TIMEOUT 120)
1156
    set_tests_properties(test_hybrid_parallel_inference_helper PROPERTIES TIMEOUT 120)
1157
    set_tests_properties(test_parallel_class_center_sample PROPERTIES TIMEOUT 120)
1158
    set_tests_properties(test_parallel_margin_cross_entropy PROPERTIES TIMEOUT 120)
1159
    set_tests_properties(test_auto_parallel_data_unshard PROPERTIES TIMEOUT 120)
1160
    set_tests_properties(test_auto_parallel_save_load PROPERTIES TIMEOUT 120)
1161
    set_tests_properties(test_auto_parallel_autoconvert PROPERTIES TIMEOUT 120)
1162
    set_tests_properties(test_collective_process_group PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1163
    set_tests_properties(test_eager_dist_api PROPERTIES TIMEOUT 100)
1164
    
1165
    if(${NCCL_VERSION} VERSION_GREATER_EQUAL 2212)
1166
        set_tests_properties(test_parallel_dygraph_sparse_embedding PROPERTIES TIMEOUT 200)
1167
        set_tests_properties(test_parallel_dygraph_transformer PROPERTIES TIMEOUT 200)
1168
        set_tests_properties(test_parallel_dygraph_sparse_embedding_over_height PROPERTIES TIMEOUT 150)
1169
    endif()
Y
YUNSHEN XIE 已提交
1170
endif()
J
Jiabin Yang 已提交
1171 1172 1173 1174 1175 1176 1177 1178

if(APPLE)
    set_tests_properties(test_imperative_transformer_sorted_gradient PROPERTIES TIMEOUT 300)
    set_tests_properties(test_multiclass_nms_op PROPERTIES TIMEOUT 300)
    set_tests_properties(test_weight_decay PROPERTIES TIMEOUT 300)
    set_tests_properties(test_imperative_static_runner_mnist PROPERTIES TIMEOUT 300)
endif()

1179
if((WITH_ROCM OR WITH_GPU) AND NOT WIN32)
Y
YUNSHEN XIE 已提交
1180
    set_tests_properties(test_collective_allgather_api PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1181
    set_tests_properties(test_collective_alltoall_api PROPERTIES TIMEOUT 120)
1182 1183
    set_tests_properties(test_collective_global_gather PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_global_scatter PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1184
    set_tests_properties(test_collective_sendrecv_api PROPERTIES TIMEOUT 120)
1185 1186
    set_tests_properties(test_collective_broadcast_api PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_allreduce_api PROPERTIES TIMEOUT 120)
1187
    if(WITH_DISTRIBUTE)
K
kuizhiqing 已提交
1188
        set_tests_properties(test_new_group_api PROPERTIES TIMEOUT 120)
1189
        set_tests_properties(test_pipeline PROPERTIES TIMEOUT 120)
1190
        set_tests_properties(test_ir_pass_pipeline PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1191
        set_tests_properties(test_static_model_parallel PROPERTIES TIMEOUT 240)
1192
        set_tests_properties(test_static_model_parallel_fused_feedforward PROPERTIES TIMEOUT 120)
1193
        set_tests_properties(test_static_model_parallel_fused_attention PROPERTIES TIMEOUT 120)
1194
        set_tests_properties(test_static_model_parallel_fused_multi_transformer PROPERTIES TIMEOUT 120)
K
kuizhiqing 已提交
1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207
        set_tests_properties(test_collective_split_embedding
            test_collective_split_embedding_none_divisible
            test_collective_split_row_linear
            test_collective_split_col_linear
            test_collective_scatter_api
            test_collective_barrier_api
            test_collective_reduce_api
            test_pipeline_parallel
            test_collective_allreduce_api
            test_new_group_api
            test_collective_broadcast_api
            test_collective_allgather_api
            test_collective_alltoall_api
1208 1209
            test_collective_global_gather
            test_collective_global_scatter
K
kuizhiqing 已提交
1210
            PROPERTIES LABELS "RUN_TYPE=DIST")
1211
    endif()
1212
    set_tests_properties(test_paddle_multiprocessing PROPERTIES TIMEOUT 120)
1213 1214 1215 1216
    set_tests_properties(test_reducescatter_api PROPERTIES TIMEOUT 120)
    set_tests_properties(test_broadcast PROPERTIES TIMEOUT 120)
    set_tests_properties(test_reducescatter PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_reduce_api PROPERTIES TIMEOUT 120)
1217
    set_tests_properties(test_pipeline_parallel PROPERTIES TIMEOUT 120)
1218 1219
    set_tests_properties(test_collective_reduce PROPERTIES TIMEOUT 120)
    set_tests_properties(test_allreduce PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1220 1221
    set_tests_properties(test_c_concat PROPERTIES TIMEOUT 120)
    set_tests_properties(test_c_split PROPERTIES TIMEOUT 120)
1222
    set_tests_properties(test_allgather PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1223
    set_tests_properties(test_c_identity PROPERTIES TIMEOUT 120)
1224 1225
    set_tests_properties(test_collective_scatter_api PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_barrier_api PROPERTIES TIMEOUT 120)
1226 1227
    set_tests_properties(test_collective_scatter PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_sendrecv PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
1228
endif()
1229
if(WITH_GPU OR WITH_ROCM)
1230
    set_tests_properties(test_imperative_auto_mixed_precision PROPERTIES TIMEOUT 300)
1231
    set_tests_properties(test_parallel_dygraph_sync_batch_norm PROPERTIES TIMEOUT 120)
1232
    set_tests_properties(test_rank_attention_op PROPERTIES TIMEOUT 120)
1233
endif()
Y
YUNSHEN XIE 已提交
1234
set_tests_properties(test_inplace_addto_strategy PROPERTIES TIMEOUT 120)
1235
set_tests_properties(test_eigvals_op PROPERTIES TIMEOUT 400)
F
From00 已提交
1236
set_tests_properties(test_tensordot PROPERTIES TIMEOUT 1000)
1237
set_tests_properties(test_tensordot PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
1238
set_tests_properties(test_cuda_memory_reserved PROPERTIES ENVIRONMENT "FLAGS_allocator_strategy=auto_growth")
X
xiongkun 已提交
1239
if (WITH_GLOO)
1240
    set_tests_properties(test_parallel_dygraph_dataparallel_cpuonly PROPERTIES TIMEOUT 30)
X
xiongkun 已提交
1241 1242 1243 1244
    set_tests_properties(test_parallel_dygraph_unused_variables_gloo PROPERTIES TIMEOUT 120)
    set_tests_properties(test_parallel_dygraph_sparse_embedding_gloo PROPERTIES TIMEOUT 120)
    set_tests_properties(test_parallel_dygraph_sparse_embedding_over_height_gloo PROPERTIES TIMEOUT 120)
endif()