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

122
if(NOT WITH_PYTHON AND ON_INFER)
123 124 125
    LIST(REMOVE_ITEM TEST_OPS test_eager_trace_op)
endif()

L
Li Min 已提交
126
if(NOT WITH_GPU)
127
    LIST(REMOVE_ITEM TEST_OPS test_fused_feedforward_op)
L
Li Min 已提交
128
    LIST(REMOVE_ITEM TEST_OPS test_fused_attention_op)
129
    LIST(REMOVE_ITEM TEST_OPS test_fused_attention_op_api)
130
    LIST(REMOVE_ITEM TEST_OPS test_fused_transformer_encoder_layer)
131 132 133 134 135 136 137 138 139 140 141
    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)
endif()

if (WITH_GPU)
    if (CUDA_VERSION LESS 11.6)
        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)
    endif()
L
Li Min 已提交
142 143
endif()

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

189 190 191
# Temporally disable test_deprecated_decorator
LIST(REMOVE_ITEM TEST_OPS test_deprecated_decorator)

192 193 194 195
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)
196 197 198 199 200 201 202
    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)
203 204
    LIST(REMOVE_ITEM TEST_OPS test_ir_memory_optimize_nlp)
    LIST(REMOVE_ITEM TEST_OPS test_nvprof)
205

206 207
    # TODO: Fix these unittests failed on Windows
    LIST(REMOVE_ITEM TEST_OPS test_debugger)
H
huangxu96 已提交
208 209 210
    if (WITH_GPU)
        LIST(REMOVE_ITEM TEST_OPS test_update_loss_scaling_op)
    endif()
211 212 213 214 215 216 217 218 219 220
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)
221
    LIST(REMOVE_ITEM TEST_OPS test_collective_cpu_barrier_with_gloo)
222 223

    # TODO: Fix these unittests failed on Windows
224
    list(REMOVE_ITEM TEST_OPS test_fake_init_op)
H
hutuxian 已提交
225 226
endif()

227 228 229 230 231
if(NOT WITH_DISTRIBUTE)
    LIST(REMOVE_ITEM TEST_OPS test_fleet_rolemaker_new)
    LIST(REMOVE_ITEM TEST_OPS test_desc_clone_dist)
endif()

232 233 234 235
if(WIN32)
    LIST(REMOVE_ITEM TEST_OPS test_complex_matmul)
endif()

G
gongweibao 已提交
236
LIST(REMOVE_ITEM TEST_OPS test_fleet_checkpoint)
237
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint)
238
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint1)
239
LIST(REMOVE_ITEM TEST_OPS test_auto_checkpoint2)
240 241 242 243 244 245
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)
246
LIST(REMOVE_ITEM TEST_OPS test_checkpoint_saver)
T
tangwei12 已提交
247

G
gongweibao 已提交
248 249
if(APPLE OR WIN32)
    LIST(REMOVE_ITEM TEST_OPS test_fs_interface)
250
    LIST(REMOVE_ITEM TEST_OPS test_fleet_metric)
G
gongweibao 已提交
251 252
endif()

S
ShenLiang 已提交
253 254
list(REMOVE_ITEM TEST_OPS test_parallel_dygraph_hybrid_parallel)

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

257 258
if (NOT WITH_GLOO)
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_dataparallel_cpuonly)
X
xiongkun 已提交
259 260 261 262

    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)
263
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_sparse_embedding_diff_length_gloo)
264 265
endif()

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

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

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

329
if(((NOT WITH_ROCM) AND (NOT WITH_GPU)) OR WIN32)
H
hutuxian 已提交
330
    LIST(REMOVE_ITEM TEST_OPS test_boxps)
331
endif()
332 333 334
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 已提交
335

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

339
if(APPLE)
340 341 342 343
    if(NOT WITH_DISTRIBUTE)
        list(REMOVE_ITEM TEST_OPS test_desc_clone)
        list(REMOVE_ITEM TEST_OPS test_program_code)
    endif(NOT WITH_DISTRIBUTE)
344
    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_*")
345 346
    # this op is not support on mac
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
347
    list(REMOVE_ITEM TEST_OPS test_detection_map_op)
348
    list(REMOVE_ITEM TEST_OPS test_fuse_elewise_add_act_pass)
349
endif()
T
Tao Luo 已提交
350 351 352 353
if(NOT WITH_MKLML)
    # this op is not support on openblas
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
endif()
354

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

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

T
tangwei12 已提交
364
list(REMOVE_ITEM TEST_OPS test_fleet_pyramid_hash)
K
Kevin 已提交
365

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

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

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

382
    if(WITH_COVERAGE AND NOT (WITH_INCREMENTAL_COVERAGE AND "$ENV{PADDLE_GIT_DIFF_PY_FILE}" STREQUAL ""))
383 384 385 386 387 388 389 390 391 392 393 394 395
        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()
396
    else()
397 398 399 400 401 402 403 404 405 406 407
        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()
408 409 410
    endif()

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

419

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

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

430

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

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

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

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

460 461 462 463 464 465 466 467 468 469 470
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 已提交
471
    set(timeout 120)
472 473 474 475 476 477 478 479
    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}
480
            COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python
481 482 483 484 485 486
            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}
487
            COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python
488 489 490 491 492 493 494 495 496 497
            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)
498
        set_tests_properties(${TARGET_NAME} PROPERTIES LABELS ${parallel_bash_test_modules_LABELS})
499 500 501
    endif()
endfunction()

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

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

541 542 543 544 545
# 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 已提交
546 547 548
if (APPLE OR WIN32)
  list(REMOVE_ITEM TEST_OPS test_dataset)
  list(REMOVE_ITEM TEST_OPS test_dataset_dataloader)
549
  list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_base)
550
  # list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception)
551
  list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_process)
552 553
  list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_fds_clear)
  list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exit_func)
554
  list(REMOVE_ITEM TEST_OPS test_imperative_signal_handler)
555 556
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_static)
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_dynamic)
557
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_exception)
558
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_iterable_dataset)
559
  list(REMOVE_ITEM TEST_OPS test_multiprocess_dataloader_dataset)
560
  list(REMOVE_ITEM TEST_OPS test_paddle_multiprocessing)
Z
Zeng Jinle 已提交
561 562
endif()

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

567 568 569 570
if(NOT WITH_GPU OR WIN32 OR APPLE)
  list(REMOVE_ITEM TEST_OPS test_build_strategy_fusion_group_pass)
endif()

571
# Some ops need to check results when gc is enabled
572
# Currently, only ops that register NoNeedBufferVarsInference need to do this test
573 574
set(TEST_OPS_WITH_GC
  test_affine_channel_op
575 576 577 578 579
  test_concat_op
  test_elementwise_add_op
  test_elementwise_sub_op
  test_fill_zeros_like2_op
  test_gather_op
580
  test_gather_nd_op
581 582 583 584 585 586 587 588
  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
589
  test_squared_l2_distance_op)
590 591 592 593 594 595

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()

596 597 598 599 600 601 602 603 604
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 已提交
605 606 607
foreach(TEST_OP ${TEST_OPS})
    py_test_modules(${TEST_OP} MODULES ${TEST_OP})
endforeach(TEST_OP)
608
py_test_modules(test_adam_op_multi_thread MODULES test_adam_op ENVS FLAGS_inner_op_parallelism=4)
609
if (WITH_GPU OR WITH_XPU OR WITH_ASCEND OR WITH_ASCEND_CL OR APPLE)
Z
zhangchunle 已提交
610 611 612
    py_test_modules(test_warpctc_op MODULES test_warpctc_op)
    set_tests_properties(test_warpctc_op PROPERTIES TIMEOUT 120)
endif()
613 614
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})
615
py_test_modules(test_imperative_resnet MODULES test_imperative_resnet ENVS
616
    FLAGS_cudnn_deterministic=1 SERIAL)
617
set_tests_properties(test_imperative_resnet PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
618
py_test_modules(test_imperative_resnet_sorted_gradient MODULES test_imperative_resnet_sorted_gradient ENVS
619
        FLAGS_cudnn_deterministic=1 SERIAL)
620
set_tests_properties(test_imperative_resnet_sorted_gradient PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
M
minqiyang 已提交
621
py_test_modules(test_imperative_mnist MODULES test_imperative_mnist ENVS
622
    FLAGS_cudnn_deterministic=1)
623
py_test_modules(test_imperative_mnist_sorted_gradient MODULES test_imperative_mnist_sorted_gradient ENVS
624
        FLAGS_cudnn_deterministic=1)
Y
Yan Xu 已提交
625
py_test_modules(test_imperative_se_resnext MODULES test_imperative_se_resnext ENVS
626
    FLAGS_cudnn_deterministic=1 SERIAL)
627
set_tests_properties(test_imperative_se_resnext PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE:NIGHTLY")
628 629
py_test_modules(test_imperative_ocr_attention_model MODULES test_imperative_ocr_attention_model ENVS
        FLAGS_cudnn_deterministic=1 SERIAL)
630 631 632
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")
633 634 635 636
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)
637
set_tests_properties(test_conv2d_op PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
638
set_tests_properties(test_faster_tokenizer_op PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
639 640
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 已提交
641 642 643
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")
644
if(WITH_DISTRIBUTE)
645
    add_subdirectory(distributed_passes)
Z
ziyoujiyi 已提交
646
    add_subdirectory(ps)
647 648
    add_subdirectory(auto_parallel)

649 650 651 652
    # FIXME(typhoonzero): add these tests back
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_transformer")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_transpiler")

653 654 655
    # TODO(sandyhouse): fix and add the ut back
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_mnist_hallreduce")

656 657 658 659
    #not need
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_base")
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_fleet_base")

T
tangwei12 已提交
660

661 662 663 664 665 666 667 668
    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 已提交
669 670
    list(REMOVE_ITEM DIST_TEST_OPS "test_dist_fleet_gloo")

T
tangwei12 已提交
671
    py_test_modules(test_recv_save_op MODULES test_recv_save_op ENVS ${dist_ENVS})
672
    py_test_modules(test_communicator_async MODULES test_communicator_async ENVS ${dist_ENVS})
T
Thunderbrook 已提交
673
    py_test_modules(test_communicator_ps_gpu MODULES test_communicator_ps_gpu ENVS ${dist_ENVS})
674 675
    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 已提交
676
    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 已提交
677
    py_test_modules(test_collective_optimizer MODULES test_collective_optimizer)
678
    if(NOT APPLE)
679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696
        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 已提交
697
        if(NOT WIN32)
698 699
            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 已提交
700
            py_test_modules(test_auto_parallel_searcher MODULES test_auto_parallel_searcher ENVS ${dist_ENVS})
C
caozhou 已提交
701
            py_test_modules(test_auto_parallel_reshard MODULES test_auto_parallel_reshard ENVS ${dist_ENVS})
702
            py_test_modules(test_auto_parallel_dist_tensor MODULES test_auto_parallel_dist_tensor ENVS ${dist_ENVS})
C
caozhou 已提交
703 704 705
            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})
706
            py_test_modules(test_auto_parallel_cost_model MODULES test_auto_parallel_cost_model ENVS ${dist_ENVS})
707 708 709 710 711 712 713 714
            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 已提交
715
        endif(NOT WIN32)
716
    endif(NOT APPLE)
G
gongweibao 已提交
717
    if(WITH_DGC)
W
WangXi 已提交
718 719
        # if with dgc, test all dgc tests.
        # NOTE. dist dgc tests is already in DIST_TEST_OPS
G
gongweibao 已提交
720
        py_test_modules(test_dgc_op MODULES test_dgc_op)
721 722
        py_test_modules(test_dgc_momentum_op MODULES test_dgc_momentum_op)
        py_test_modules(test_dgc_optimizer MODULES test_dgc_optimizer)
723
        py_test_modules(test_fleet_dgc_meta_optimizer MODULES test_fleet_dgc_meta_optimizer)
W
WangXi 已提交
724 725 726 727
    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 已提交
728
    endif()
J
JiabinYang 已提交
729
    if(NOT APPLE)
730
        if(WITH_GPU OR WITH_ROCM)
731
            bash_test_modules(test_c_comm_init_op START_BASH test_c_comm_init_op.sh  ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
732
            py_test_modules(test_launch_coverage MODULES test_launch_coverage)
W
WangXi 已提交
733
        endif()
734

735
        bash_test_modules(test_fleetrun START_BASH test_fleetrun.sh ENVS PADDLE_BINARY_DIR=${PADDLE_BINARY_DIR})
736 737
        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})
738 739 740
            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})
741
        endif()
742
        if(WITH_ASCEND OR WITH_ASCEND_CL)
743 744 745
            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()
746

747
        # port range (20000, 23000) is reserved for dist-ops
G
gongweibao 已提交
748
        set(dist_ut_port 20001)
749
        foreach(TEST_OP ${DIST_TEST_OPS})
750
            bash_test_modules(${TEST_OP} START_BASH dist_test.sh SERIAL LABELS "RUN_TYPE=EXCLUSIVE" ENVS "PADDLE_DIST_UT_PORT=${dist_ut_port}")
751
            MATH(EXPR dist_ut_port "${dist_ut_port}+30")
752 753 754
            if(dist_ut_port GREATER_EQUAL 22998)
                message(FATAL_ERROR "available ports have been exhausted:${dist_ut_port}")
            endif()
755
        endforeach(TEST_OP)
756
        # solve it later.
757
        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 已提交
758 759 760
        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()
761 762 763
        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()
764
    endif(NOT APPLE)
765
endif()
766

767
py_test_modules(test_parallel_executor_crf MODULES test_parallel_executor_crf)
768 769
# 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.
770
# We guess there are some bugs in cuda 10.1 or 10.2,
771 772 773 774 775 776
# 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()
777
py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer)
778 779
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 已提交
780 781
    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)
782
    py_test_modules(test_fetch_lod_tensor_array MODULES test_fetch_lod_tensor_array ENVS CUDA_VISIBLE_DEVICES=0)
783 784
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 已提交
785 786
    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)
787
    py_test_modules(test_fetch_lod_tensor_array MODULES test_fetch_lod_tensor_array)
788
endif()
789

H
hutuxian 已提交
790
py_test_modules(test_data_norm_op MODULES test_data_norm_op)
Z
Zhen Wang 已提交
791
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 已提交
792
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)
793

794
# NOTE: These unittests will appear NaN steadily in windows CI. After analysis,
795 796
# 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
797 798
# inconsistent with that in non-inference mode.
if(NOT ON_INFER)
799 800 801 802
    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)
803
    set_tests_properties(test_parallel_executor_seresnext_base_cpu PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
804
    set_tests_properties(test_parallel_executor_seresnext_with_reduce_cpu PROPERTIES TIMEOUT 750)
805
    set_tests_properties(test_parallel_executor_seresnext_with_reduce_cpu PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
806
    set_tests_properties(test_parallel_executor_seresnext_with_fuse_all_reduce_cpu PROPERTIES TIMEOUT 750)
807
    set_tests_properties(test_parallel_executor_seresnext_with_fuse_all_reduce_cpu PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
808
endif()
809

810 811 812
if(NOT WIN32)
    # TODO: fix these unittests failure on Windows
    py_test_modules(test_layers MODULES test_layers ENVS FLAGS_cudnn_deterministic=1)
813
    py_test_modules(test_ir_memory_optimize_transformer MODULES test_ir_memory_optimize_transformer)
814 815 816
    # 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 已提交
817
endif()
818

819
if(WITH_DISTRIBUTE AND NOT APPLE AND NOT WIN32)
G
gongweibao 已提交
820 821 822 823 824 825 826 827 828 829 830 831
    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")
832 833
endif()

834
add_subdirectory(sequence)
835
add_subdirectory(dygraph_to_static)
F
Feiyu Chan 已提交
836
add_subdirectory(rnn)
837
add_subdirectory(autograd)
838
add_subdirectory(distribution)
839 840 841 842

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

T
tianshuo78520a 已提交
844 845 846 847
if (WITH_XPU)
    add_subdirectory(xpu)
endif()

848
# dist xpu tests:
849 850 851 852 853
if (WITH_XPU_BKCL)
    py_test(test_collective_reduce_api_xpu SRCS "test_collective_reduce_api.py")
    py_test(test_collective_allreduce_api_xpu SRCS "test_collective_allreduce_api.py")
endif()

854 855 856 857
if(WIN32)
    cc_test(cc_imp_py_test SRCS cc_imp_py_test.cc DEPS python)
endif()

858 859 860 861
if (WITH_ASCEND_CL)
    add_subdirectory(npu)
endif()

862 863 864
if (WITH_MKLDNN)
    add_subdirectory(mkldnn)
endif()
865

J
jianghaicheng 已提交
866 867 868 869
if (WITH_IPU)
    add_subdirectory(ipu)
endif()

870 871 872 873
if (WITH_MLU)
    add_subdirectory(mlu)
endif()

874 875
add_subdirectory(asp)

876 877
add_subdirectory(ir)

H
hong 已提交
878 879
add_subdirectory(interpreter)

880
if (WITH_TESTING)
Z
Zeng Jinle 已提交
881
    set_property(TEST test_parallel_executor_mnist PROPERTY ENVIRONMENT GLOG_vmodule=all_reduce_deps_pass=10)
Z
Zeng Jinle 已提交
882
    set_property(TEST test_parallel_executor_fix_op_run_order PROPERTY ENVIRONMENT GLOG_vmodule=fix_op_run_order_pass=10)
883 884
endif()

885
set_tests_properties(test_parallel_executor_test_while_train test_parallel_executor_mnist
886
        test_parallel_executor_feed_persistable_var
887
        test_buffer_shared_memory_reuse_pass_and_fuse_optimization_op_pass
888
        test_data_norm_op
889
        test_dataloader_keep_order
890 891
        test_dataloader_unkeep_order
        test_parallel_executor_inference_feed_partial_data
892
        test_parallel_ssa_graph_inference_feed_partial_data
893
        test_fetch_unmerged
894
        test_buffer_shared_memory_reuse_pass PROPERTIES LABELS "RUN_TYPE=DIST")
895 896
# disable test_parallel_executor_fetch_isolated_var
# set_tests_properties(test_parallel_executor_fetch_isolated_var PROPERTIES LABELS "RUN_TYPE=DIST")
K
Kaipeng Deng 已提交
897
set_tests_properties(test_parallel_executor_crf test_sync_batch_norm_op test_inplace_abn_op
898 899 900
        test_parallel_executor_seresnext_base_gpu
        test_parallel_executor_seresnext_with_reduce_gpu
        test_parallel_executor_seresnext_with_fuse_all_reduce_gpu
901 902
        test_distributed_fused_lamb_op_with_clip
        test_distributed_fused_lamb_op_without_clip
903
        test_parallel_executor_fetch_isolated_var
C
chalsliu 已提交
904
        PROPERTIES LABELS "RUN_TYPE=DIST")
905

906
if(NOT WIN32 AND NOT APPLE)
907
    set_tests_properties(test_imperative_signal_handler PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
C
chalsliu 已提交
908 909 910 911 912 913
    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")
914 915
    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")
916
    set_tests_properties(test_multiprocess_dataloader_dataset PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
917
    set_tests_properties(test_multiprocess_dataloader_static PROPERTIES TIMEOUT 120)
918
endif()
Y
YUNSHEN XIE 已提交
919

920
if (NOT WIN32)
921
    set_tests_properties(test_multiprocess_reader_exception PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
922
    set_tests_properties(test_layers PROPERTIES TIMEOUT 120)
F
feng_shuai 已提交
923 924 925
    if (WITH_NV_JETSON)
    set_tests_properties(test_ir_memory_optimize_transformer PROPERTIES TIMEOUT 1200)
    else ()
926
    set_tests_properties(test_ir_memory_optimize_transformer PROPERTIES TIMEOUT 120)
F
feng_shuai 已提交
927
    endif ()
928 929 930
endif()

if (WITH_DISTRIBUTE AND NOT WIN32)
931
    set_tests_properties(test_fleet_utils PROPERTIES TIMEOUT 120)
932
    set_tests_properties(test_collective_cpu_barrier_with_gloo PROPERTIES TIMEOUT 40)
933 934 935 936
endif()

if (WITH_DISTRIBUTE)
    set_tests_properties(test_communicator_half_async PROPERTIES TIMEOUT 120)
937 938
    set_tests_properties(test_dist_fleet_ctr2 PROPERTIES TIMEOUT 200)
    set_tests_properties(test_dist_fleet_sparse_embedding_ctr PROPERTIES TIMEOUT 200)
939
    set_tests_properties(test_dist_fleet_infer PROPERTIES TIMEOUT 200)
940
    set_tests_properties(test_dist_fleet_raw_program_optimizer PROPERTIES TIMEOUT 120)
941
    set_tests_properties(test_dist_fleet_raw_program_optimizer_fuse_allreduce PROPERTIES TIMEOUT 60)
942
endif()
943

944
if (WITH_DISTRIBUTE AND NOT APPLE)
945
    if(WITH_GPU OR WITH_ROCM)
946
        set_tests_properties(test_c_comm_init_op PROPERTIES TIMEOUT 120)
947
        set_tests_properties(test_dist_mnist_gradient_merge PROPERTIES TIMEOUT 160)
948
    endif()
Y
YUNSHEN XIE 已提交
949 950 951
endif()

# setting timeout value as 15S
952
set_tests_properties(test_run PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
953 954
set_tests_properties(test_sync_batch_norm_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_cross_op PROPERTIES TIMEOUT 120)
955
set_tests_properties(test_imperative_lod_tensor_to_selected_rows PROPERTIES TIMEOUT 200)
Y
YUNSHEN XIE 已提交
956 957
set_tests_properties(test_lstm_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_imperative_star_gan_with_gradient_penalty PROPERTIES TIMEOUT 120)
Z
zhangchunle 已提交
958

Y
YUNSHEN XIE 已提交
959 960 961 962
set_tests_properties(test_bicubic_interp_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_deformable_conv_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_nearest_interp_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_profiler PROPERTIES TIMEOUT 120)
963
set_tests_properties(test_inplace_eager_fluid PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
964 965 966 967
set_tests_properties(test_inplace_softmax_with_cross_entropy PROPERTIES TIMEOUT 120)
set_tests_properties(test_cross_entropy2_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_fetch_unmerged PROPERTIES TIMEOUT 120)
set_tests_properties(test_gru_unit_op PROPERTIES TIMEOUT 120)
D
Double_V 已提交
968
set_tests_properties(test_activation_nn_grad PROPERTIES TIMEOUT 200)
Y
YUNSHEN XIE 已提交
969 970 971
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)
972
set_tests_properties(test_regularizer_api PROPERTIES TIMEOUT 150)
Y
YUNSHEN XIE 已提交
973
set_tests_properties(test_multiclass_nms_op PROPERTIES TIMEOUT 120)
974
if(NOT WIN32)
F
feng_shuai 已提交
975 976 977
    if (WITH_NV_JETSON)
    set_tests_properties(test_ir_memory_optimize_nlp PROPERTIES TIMEOUT 1200)
    else ()
978
    set_tests_properties(test_ir_memory_optimize_nlp PROPERTIES TIMEOUT 120)
F
feng_shuai 已提交
979
    endif ()
980
endif()
Y
YUNSHEN XIE 已提交
981 982 983 984 985 986 987 988 989 990 991
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)
992
set_tests_properties(test_static_save_load PROPERTIES TIMEOUT 250)
993
set_tests_properties(test_pylayer_op PROPERTIES TIMEOUT 120)
994
set_tests_properties(test_paddle_save_load_binary PROPERTIES TIMEOUT 120)
W
WeiXin 已提交
995
if (WIN32)
W
WeiXin 已提交
996
    set_tests_properties(test_static_save_load_large PROPERTIES TIMEOUT 900)
W
WeiXin 已提交
997 998
    set_tests_properties(test_paddle_save_load PROPERTIES TIMEOUT 250)
else()
W
WeiXin 已提交
999
    set_tests_properties(test_static_save_load_large PROPERTIES TIMEOUT 600)
1000
    set_tests_properties(test_paddle_save_load PROPERTIES TIMEOUT 250)
W
WeiXin 已提交
1001
endif()
F
feng_shuai 已提交
1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018
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()
1019
set_tests_properties(test_imperative_selected_rows_to_lod_tensor PROPERTIES TIMEOUT 200)
Y
YUNSHEN XIE 已提交
1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032
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)
set_tests_properties(test_nn_grad PROPERTIES TIMEOUT 120)
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)
1033 1034
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)
Y
YUNSHEN XIE 已提交
1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047
set_tests_properties(test_elementwise_min_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_nan_inf PROPERTIES TIMEOUT 120)
set_tests_properties(test_deformable_conv_v1_op PROPERTIES TIMEOUT 120)
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 已提交
1048
set_tests_properties(test_conv2d_transpose_op PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
1049 1050 1051 1052 1053
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 已提交
1054
set_tests_properties(test_svd_op PROPERTIES TIMEOUT 80)
1055
set_tests_properties(test_qr_op PROPERTIES TIMEOUT 60)
Y
YUNSHEN XIE 已提交
1056 1057 1058 1059 1060
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)
1061
set_tests_properties(test_imperative_optimizer_v2 PROPERTIES TIMEOUT 150)
Y
YUNSHEN XIE 已提交
1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084
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)
set_tests_properties(test_norm_nn_grad PROPERTIES TIMEOUT 120)
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)
set_tests_properties(test_softmax_with_cross_entropy_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_reduce_op PROPERTIES TIMEOUT 120)
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)
set_tests_properties(test_imperative_optimizer PROPERTIES TIMEOUT 120)
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 已提交
1085
set_tests_properties(test_activation_op PROPERTIES TIMEOUT 270)
Y
YUNSHEN XIE 已提交
1086 1087 1088 1089 1090 1091
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)
1092 1093
set_tests_properties(test_conv2d_op_depthwise_conv PROPERTIES TIMEOUT 120)
set_tests_properties(test_conv2d_api PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
1094 1095 1096 1097 1098 1099
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)
1100
set_tests_properties(test_fused_elemwise_activation_op PROPERTIES TIMEOUT 270)
1101
set_tests_properties(test_fused_elemwise_activation_op PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
1102 1103
set_tests_properties(test_gru_op PROPERTIES TIMEOUT 200)
set_tests_properties(test_regularizer PROPERTIES TIMEOUT 150)
Y
YUNSHEN XIE 已提交
1104 1105 1106 1107 1108
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)
1109 1110 1111 1112 1113 1114 1115
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)
1116 1117
set_tests_properties(test_reader_reset PROPERTIES TIMEOUT 120)
set_tests_properties(test_pool3d_api PROPERTIES TIMEOUT 120)
H
hlygit66666 已提交
1118
set_tests_properties(test_cumprod_op PROPERTIES TIMEOUT 120)
1119
set_tests_properties(test_split_program PROPERTIES TIMEOUT 120)
1120
if(WITH_DISTRIBUTE AND WITH_GPU AND WITH_NCCL)
1121
    set_tests_properties(test_parallel_dygraph_dataparallel PROPERTIES TIMEOUT 120)
1122
    set_tests_properties(test_parallel_dygraph_unused_variables PROPERTIES TIMEOUT 150)
1123
    set_tests_properties(test_parallel_dygraph_control_flow PROPERTIES TIMEOUT 200)
1124
    set_tests_properties(test_parallel_dygraph_no_sync PROPERTIES TIMEOUT 150)
1125
    set_tests_properties(test_parallel_dygraph_no_sync_gradient_check PROPERTIES TIMEOUT 30)
1126
    set_tests_properties(test_parallel_dygraph_pipeline_parallel PROPERTIES TIMEOUT 200)
1127
    set_tests_properties(test_parallel_dygraph_tensor_parallel PROPERTIES TIMEOUT 200)
J
JZ-LIANG 已提交
1128
    set_tests_properties(test_parallel_dygraph_sharding_parallel PROPERTIES TIMEOUT 120)
1129
    set_tests_properties(test_dygraph_sharding_optimizer_stage2 PROPERTIES TIMEOUT 120)
B
Baibaifan 已提交
1130
    set_tests_properties(test_dygraph_sharding_stage2 PROPERTIES TIMEOUT 120)
B
Baibaifan 已提交
1131
    set_tests_properties(test_dygraph_sharding_stage3 PROPERTIES TIMEOUT 120)
B
Baibaifan 已提交
1132
    set_tests_properties(test_dygraph_group_sharded_api PROPERTIES TIMEOUT 120)
1133
    set_tests_properties(test_auto_parallel_parallelizer PROPERTIES TIMEOUT 120)
1134
    set_tests_properties(test_parallel_dygraph_mp_layers PROPERTIES TIMEOUT 120)
1135
    set_tests_properties(test_hybrid_parallel_inference_helper PROPERTIES TIMEOUT 120)
1136
    set_tests_properties(test_parallel_class_center_sample PROPERTIES TIMEOUT 120)
1137
    set_tests_properties(test_parallel_margin_cross_entropy PROPERTIES TIMEOUT 120)
1138
    set_tests_properties(test_auto_parallel_data_unshard PROPERTIES TIMEOUT 120)
1139
    set_tests_properties(test_auto_parallel_save_load PROPERTIES TIMEOUT 120)
1140
    set_tests_properties(test_auto_parallel_autoconvert PROPERTIES TIMEOUT 120)
1141 1142
    set_tests_properties(test_collective_process_group PROPERTIES TIMEOUT 120)
    
1143 1144 1145 1146
    if(${NCCL_VERSION} VERSION_GREATER_EQUAL 2212)
        set_tests_properties(test_parallel_dygraph_sparse_embedding PROPERTIES TIMEOUT 120)
        set_tests_properties(test_parallel_dygraph_transformer PROPERTIES TIMEOUT 120)
    endif()
Y
YUNSHEN XIE 已提交
1147
endif()
1148
if((WITH_ROCM OR WITH_GPU) AND NOT WIN32)
Y
YUNSHEN XIE 已提交
1149
    set_tests_properties(test_collective_allgather_api PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1150
    set_tests_properties(test_collective_alltoall_api PROPERTIES TIMEOUT 120)
1151 1152
    set_tests_properties(test_collective_global_gather PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_global_scatter PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1153
    set_tests_properties(test_collective_sendrecv_api PROPERTIES TIMEOUT 120)
1154 1155
    set_tests_properties(test_collective_broadcast_api PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_allreduce_api PROPERTIES TIMEOUT 120)
1156
    if(WITH_DISTRIBUTE)
K
kuizhiqing 已提交
1157
        set_tests_properties(test_new_group_api PROPERTIES TIMEOUT 120)
1158
        set_tests_properties(test_pipeline PROPERTIES TIMEOUT 120)
1159
        set_tests_properties(test_ir_pass_pipeline PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1160
        set_tests_properties(test_static_model_parallel PROPERTIES TIMEOUT 240)
1161
        set_tests_properties(test_static_model_parallel_fused_feedforward PROPERTIES TIMEOUT 120)
1162
        set_tests_properties(test_static_model_parallel_fused_attention PROPERTIES TIMEOUT 120)
K
kuizhiqing 已提交
1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175
        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
1176 1177
            test_collective_global_gather
            test_collective_global_scatter
K
kuizhiqing 已提交
1178
            PROPERTIES LABELS "RUN_TYPE=DIST")
1179
    endif()
1180
    set_tests_properties(test_paddle_multiprocessing PROPERTIES TIMEOUT 120)
1181 1182 1183 1184
    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)
1185
    set_tests_properties(test_pipeline_parallel PROPERTIES TIMEOUT 120)
1186 1187
    set_tests_properties(test_collective_reduce PROPERTIES TIMEOUT 120)
    set_tests_properties(test_allreduce PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1188 1189
    set_tests_properties(test_c_concat PROPERTIES TIMEOUT 120)
    set_tests_properties(test_c_split PROPERTIES TIMEOUT 120)
1190
    set_tests_properties(test_allgather PROPERTIES TIMEOUT 120)
L
lilong12 已提交
1191
    set_tests_properties(test_c_identity PROPERTIES TIMEOUT 120)
1192 1193
    set_tests_properties(test_collective_scatter_api PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_barrier_api PROPERTIES TIMEOUT 120)
1194 1195
    set_tests_properties(test_collective_scatter PROPERTIES TIMEOUT 120)
    set_tests_properties(test_collective_sendrecv PROPERTIES TIMEOUT 120)
Y
YUNSHEN XIE 已提交
1196
endif()
1197
if(WITH_GPU OR WITH_ROCM)
1198
    set_tests_properties(test_imperative_auto_mixed_precision PROPERTIES TIMEOUT 300)
1199
    set_tests_properties(test_parallel_dygraph_sync_batch_norm PROPERTIES TIMEOUT 120)
1200
    set_tests_properties(test_rank_attention_op PROPERTIES TIMEOUT 120)
1201
endif()
Y
YUNSHEN XIE 已提交
1202
set_tests_properties(test_inplace_addto_strategy PROPERTIES TIMEOUT 120)
1203
set_tests_properties(test_eigvals_op PROPERTIES TIMEOUT 400)
F
From00 已提交
1204
set_tests_properties(test_tensordot PROPERTIES TIMEOUT 1000)
1205
set_tests_properties(test_tensordot PROPERTIES LABELS "RUN_TYPE=NIGHTLY")
X
xiongkun 已提交
1206 1207 1208 1209 1210
if (WITH_GLOO)
    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()