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

if(NOT WITH_DISTRIBUTE)
    list(REMOVE_ITEM TEST_OPS test_recv_op)
7 8 9 10
    list(REMOVE_ITEM TEST_OPS test_dist_transpiler)
    list(REMOVE_ITEM TEST_OPS test_simple_dist_transpiler)
    list(REMOVE_ITEM TEST_OPS test_listen_and_serv_op)
    LIST(REMOVE_ITEM TEST_OPS test_dist_mnist)
11 12
    LIST(REMOVE_ITEM TEST_OPS test_dist_mnist_nccl)
    LIST(REMOVE_ITEM TEST_OPS test_dist_mnist_lars)
13
    LIST(REMOVE_ITEM TEST_OPS test_dist_word2vec)
M
minqiyang 已提交
14
    LIST(REMOVE_ITEM TEST_OPS test_dist_ctr)
M
minqiyang 已提交
15 16 17
    LIST(REMOVE_ITEM TEST_OPS test_dist_simnet_bow)
    LIST(REMOVE_ITEM TEST_OPS test_dist_mnist_batch_merge)
    LIST(REMOVE_ITEM TEST_OPS test_dist_text_classification)
T
tangwei12 已提交
18 19
    LIST(REMOVE_ITEM TEST_OPS test_nce_remote_table_op)
    LIST(REMOVE_ITEM TEST_OPS test_hsigmoid_remote_table_op)
T
tangwei12 已提交
20
    LIST(REMOVE_ITEM TEST_OPS test_dist_fleet_ctr)
21 22
endif(NOT WITH_DISTRIBUTE)

G
gongweibao 已提交
23 24
LIST(REMOVE_ITEM TEST_OPS test_launch)

Y
Yu Yang 已提交
25 26
if (NOT ${WITH_GPU})
    LIST(REMOVE_ITEM TEST_OPS test_conv2d_fusion_op)
27
    LIST(REMOVE_ITEM TEST_OPS test_parallel_dygraph_mnist) # TODO(Yancey1989): parallel dygraph support CPU device in future
P
peizhilin 已提交
28
elseif(${CUDNN_VERSION} VERSION_LESS 7100)
D
Dang Qingqing 已提交
29 30 31
    LIST(REMOVE_ITEM TEST_OPS test_conv2d_fusion_op)
endif()

32
list(REMOVE_ITEM TEST_OPS test_seq_concat_op) # FIXME(helin): https://github.com/PaddlePaddle/Paddle/issues/8290
Q
Qiyang Min 已提交
33
list(REMOVE_ITEM TEST_OPS test_modified_huber_loss_op) # FIXME(qijun) https://github.com/PaddlePaddle/Paddle/issues/5184
34 35 36
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 已提交
37

38
list(REMOVE_ITEM TEST_OPS op_test) # op_test is a helper python file, not a test
39
list(REMOVE_ITEM TEST_OPS decorator_helper) # decorator_helper is a helper python file, not a test
40
if(APPLE)
41 42 43 44
    if(NOT WITH_DISTRIBUTE)
        list(REMOVE_ITEM TEST_OPS test_desc_clone)
        list(REMOVE_ITEM TEST_OPS test_program_code)
    endif(NOT WITH_DISTRIBUTE)
45
    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")
46 47
    # this op is not support on mac
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
48 49 50
    # TODO: add the unitest back when it fixed
    list(REMOVE_ITEM TEST_OPS test_detection_map_op)
    list(REMOVE_ITEM TEST_OPS test_dist_se_resnext)
51 52
    # TODO(tangwei12): add the unitest back when it fixed
    list(REMOVE_ITEM TEST_OPS test_dist_word2vec)
53
    list(REMOVE_ITEM TEST_OPS test_fuse_elewise_add_act_pass)
54
endif()
T
Tao Luo 已提交
55 56 57 58
if(NOT WITH_MKLML)
    # this op is not support on openblas
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
endif()
59 60 61

function(py_test_modules TARGET_NAME)
  if(WITH_TESTING)
62
    set(options SERIAL)
63
    set(oneValueArgs "")
64
    set(multiValueArgs MODULES DEPS ENVS)
65 66
    cmake_parse_arguments(py_test_modules "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
    add_test(NAME ${TARGET_NAME}
67 68 69
        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})
70 71 72
    if (py_test_modules_SERIAL)         
        set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)            
    endif()
T
tianshuo78520a 已提交
73
    set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 350)
74 75
  endif()
endfunction()
G
gongweibao 已提交
76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98

function(bash_test_modules TARGET_NAME)
    if(NOT WITH_TESTING)
        return()
    endif()

    set(options SERIAL)
    set(oneValueArgs "")
    set(multiValueArgs MODULES DEPS ENVS)
    cmake_parse_arguments(bash_test_modules "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})

    message(STATUS "CMAKE_CURRENT_BINARY_DIR:" ${CMAKE_CURRENT_BINARY_DIR})

    add_test(NAME ${TARGET_NAME}
        COMMAND ${CMAKE_COMMAND} -E env PYTHONPATH=${PADDLE_BINARY_DIR}/python ${bash_test_modules_ENVS}
        bash ${CMAKE_CURRENT_BINARY_DIR}/${bash_test_modules_MODULES}
        WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
    if (bash_test_modules_SERIAL)
        set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)
    endif()
    set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 600)
endfunction()

99
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
T
typhoonzero 已提交
100
list(REMOVE_ITEM TEST_OPS test_dist_train)
T
typhoonzero 已提交
101
list(REMOVE_ITEM TEST_OPS test_dist_transpiler)
Y
yuyang18 已提交
102
list(REMOVE_ITEM TEST_OPS test_parallel_executor_crf)
Y
yuyang18 已提交
103
list(REMOVE_ITEM TEST_OPS test_parallel_executor_fetch_feed)
104
list(REMOVE_ITEM TEST_OPS test_dist_se_resnext)
105
list(REMOVE_ITEM TEST_OPS test_dgc_op)
106
list(REMOVE_ITEM TEST_OPS test_dist_se_resnext_nccl)
X
Xin Pan 已提交
107
list(REMOVE_ITEM TEST_OPS test_dist_transformer)
108
list(REMOVE_ITEM TEST_OPS test_parallel_executor_transformer)
D
dengkaipeng 已提交
109 110
list(REMOVE_ITEM TEST_OPS test_bilinear_interp_op)
list(REMOVE_ITEM TEST_OPS test_nearest_interp_op)
111
list(REMOVE_ITEM TEST_OPS test_imperative_resnet)
112 113
list(REMOVE_ITEM TEST_OPS test_imperative_resnet_sorted_gradient)
list(REMOVE_ITEM TEST_OPS test_imperative_mnist_sorted_gradient)
Y
Yan Xu 已提交
114
list(REMOVE_ITEM TEST_OPS test_imperative_se_resnext)
M
minqiyang 已提交
115
list(REMOVE_ITEM TEST_OPS test_imperative_mnist)
D
dzhwinter 已提交
116
list(REMOVE_ITEM TEST_OPS test_ir_memory_optimize_transformer)
L
lujun 已提交
117
list(REMOVE_ITEM TEST_OPS test_layers)
118
list(REMOVE_ITEM TEST_OPS test_imperative_ocr_attention_model)
119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154

# Some ops need to check results when gc is enabled
# Currently, only ops that register NoNeedBufferVarsInference need to do this test   
set(TEST_OPS_WITH_GC 
  test_concat_op
  test_elementwise_add_op
  test_elementwise_sub_op
  test_fill_constant_batch_size_like_op
  test_fill_zeros_like2_op
  test_gather_op
  test_gaussian_random_batch_size_like_op
  test_linear_chain_crf_op
  test_lod_reset_op
  test_lookup_table_op
  test_mean_op
  test_pad2d_op
  test_scatter_op
  test_sequence_concat
  test_seq_conv
  test_seq_pool
  test_sequence_expand_as
  test_sequence_expand
  test_sequence_pad_op
  test_sequence_unpad_op
  test_sequence_scatter_op
  test_sequence_slice_op
  test_slice_op
  test_space_to_depth_op
  test_squared_l2_distance_op
  test_uniform_random_batch_size_like_op)

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

Y
yuyang18 已提交
155 156 157
foreach(TEST_OP ${TEST_OPS})
    py_test_modules(${TEST_OP} MODULES ${TEST_OP})
endforeach(TEST_OP)
158
py_test_modules(test_adam_op_multi_thread MODULES test_adam_op ENVS FLAGS_inner_op_parallelism=4)
159 160 161
py_test_modules(test_warpctc_op MODULES test_warpctc_op)
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})
162
py_test_modules(test_imperative_resnet MODULES test_imperative_resnet ENVS
163
    FLAGS_cudnn_deterministic=1 SERIAL)
164
set_tests_properties(test_imperative_resnet PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
165
py_test_modules(test_imperative_resnet_sorted_gradient MODULES test_imperative_resnet_sorted_gradient ENVS
166
        FLAGS_cudnn_deterministic=1 SERIAL)
167
set_tests_properties(test_imperative_resnet_sorted_gradient PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
M
minqiyang 已提交
168
py_test_modules(test_imperative_mnist MODULES test_imperative_mnist ENVS
169
    FLAGS_cudnn_deterministic=1)
170
py_test_modules(test_imperative_mnist_sorted_gradient MODULES test_imperative_mnist_sorted_gradient ENVS
171
        FLAGS_cudnn_deterministic=1)
Y
Yan Xu 已提交
172
py_test_modules(test_imperative_se_resnext MODULES test_imperative_se_resnext ENVS
173 174 175
    FLAGS_cudnn_deterministic=1 SERIAL)
set_tests_properties(test_imperative_se_resnext PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")

176
if(WITH_DISTRIBUTE)
177
    py_test_modules(test_dist_train MODULES test_dist_train)
178
    set_tests_properties(test_listen_and_serv_op PROPERTIES TIMEOUT 20)
G
gongweibao 已提交
179 180 181
    if(WITH_DGC)
        py_test_modules(test_dgc_op MODULES test_dgc_op)
    endif()
J
JiabinYang 已提交
182
    if(NOT APPLE)
183
        set_tests_properties(test_dist_mnist PROPERTIES TIMEOUT 200)
184
        set_tests_properties(test_dist_mnist_nccl PROPERTIES TIMEOUT 250)
185
        set_tests_properties(test_dist_mnist_lars PROPERTIES TIMEOUT 200)
186
        set_tests_properties(test_dist_word2vec PROPERTIES TIMEOUT 200)
187
        py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext)
188
        py_test_modules(test_dist_se_resnext_nccl MODULES test_dist_se_resnext_nccl)
G
gongweibao 已提交
189
        bash_test_modules(test_launch MODULES test_launch.sh)
190
        # FIXME(typhoonzero): add these tests back
191 192
        # py_test_modules(test_dist_transformer MODULES test_dist_transformer)
        # set_tests_properties(test_dist_transformer PROPERTIES TIMEOUT 1000)
193
    endif(NOT APPLE)
194
    # py_test_modules(test_dist_transpiler MODULES test_dist_transpiler)
195
endif()
196

197 198
py_test_modules(test_parallel_executor_crf MODULES test_parallel_executor_crf)
py_test_modules(test_parallel_executor_fetch_feed MODULES test_parallel_executor_fetch_feed)
C
chengduo 已提交
199
set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 450)
200
set_tests_properties(test_parallel_executor_seresnext PROPERTIES TIMEOUT 740)
201
py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer)
202
py_test_modules(test_layers MODULES test_layers ENVS FLAGS_cudnn_deterministic=1)
D
dzhwinter 已提交
203
if(NOT WIN32)
204
    py_test_modules(test_ir_memory_optimize_transformer MODULES test_ir_memory_optimize_transformer)
D
dzhwinter 已提交
205
endif()
206

207
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
C
chengduo 已提交
208 209
    # change the timeout from 600 to 2200, because in debug mode, this test need more time.
    set_tests_properties(test_parallel_executor_seresnext PROPERTIES TIMEOUT 2200)
210
endif()
211 212 213 214

if (WITH_NGRAPH)
    add_subdirectory(ngraph)
endif()
215 216 217 218

if (WITH_MKLDNN)
    add_subdirectory(mkldnn)
endif()
219 220 221

if(WITH_DISTRIBUTE)
    set_tests_properties(test_listen_and_serv_op test_nce_remote_table_op test_hsigmoid_remote_table_op
222
            PROPERTIES LABELS "RUN_TYPE=DIST")
223 224
endif()

225
set_tests_properties(test_recordio_reader test_parallel_executor_test_while_train test_parallel_executor_mnist
226
        test_parallel_executor_seresnext test_parallel_executor_crf test_sync_batch_norm_op
227
        PROPERTIES LABELS "RUN_TYPE=DIST")