CMakeLists.txt 6.9 KB
Newer Older
1 2 3 4 5
file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")

if(NOT WITH_DISTRIBUTE)
    list(REMOVE_ITEM TEST_OPS test_recv_op)
6 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)
    LIST(REMOVE_ITEM TEST_OPS test_dist_word2vec)
M
minqiyang 已提交
11
    LIST(REMOVE_ITEM TEST_OPS test_dist_ctr)
M
minqiyang 已提交
12 13 14
    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 已提交
15 16
    LIST(REMOVE_ITEM TEST_OPS test_nce_remote_table_op)
    LIST(REMOVE_ITEM TEST_OPS test_hsigmoid_remote_table_op)
17 18
endif(NOT WITH_DISTRIBUTE)

Y
Yu Yang 已提交
19 20
if (NOT ${WITH_GPU})
    LIST(REMOVE_ITEM TEST_OPS test_conv2d_fusion_op)
P
peizhilin 已提交
21
elseif(${CUDNN_VERSION} VERSION_LESS 7100)
D
Dang Qingqing 已提交
22 23 24
    LIST(REMOVE_ITEM TEST_OPS test_conv2d_fusion_op)
endif()

25
list(REMOVE_ITEM TEST_OPS test_seq_concat_op) # FIXME(helin): https://github.com/PaddlePaddle/Paddle/issues/8290
Q
Qiyang Min 已提交
26
list(REMOVE_ITEM TEST_OPS test_modified_huber_loss_op) # FIXME(qijun) https://github.com/PaddlePaddle/Paddle/issues/5184
27 28 29 30 31
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

list(REMOVE_ITEM TEST_OPS op_test) # op_test is a helper python file, not a test
list(REMOVE_ITEM TEST_OPS decorators) # decorators is a helper python file, not a test
32
if(APPLE)
33 34 35 36
    if(NOT WITH_DISTRIBUTE)
        list(REMOVE_ITEM TEST_OPS test_desc_clone)
        list(REMOVE_ITEM TEST_OPS test_program_code)
    endif(NOT WITH_DISTRIBUTE)
37
    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")
38 39
    # this op is not support on mac
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
40 41 42
    # 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)
43
    list(REMOVE_ITEM TEST_OPS test_fuse_elewise_add_act_pass)
44
endif()
T
Tao Luo 已提交
45 46 47 48
if(NOT WITH_MKLML)
    # this op is not support on openblas
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
endif()
49 50 51

function(py_test_modules TARGET_NAME)
  if(WITH_TESTING)
Y
yuyang18 已提交
52
    set(options SERIAL)
53
    set(oneValueArgs "")
54
    set(multiValueArgs MODULES DEPS ENVS)
55 56
    cmake_parse_arguments(py_test_modules "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
    add_test(NAME ${TARGET_NAME}
57 58 59
        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})
Y
yuyang18 已提交
60
    if (py_test_modules_SERIAL)
X
Xin Pan 已提交
61
        set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)
Y
yuyang18 已提交
62
    endif()
X
Xin Pan 已提交
63
    set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 600)
64 65 66
  endif()
endfunction()
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
T
typhoonzero 已提交
67
list(REMOVE_ITEM TEST_OPS test_dist_train)
T
typhoonzero 已提交
68
list(REMOVE_ITEM TEST_OPS test_dist_transpiler)
Y
yuyang18 已提交
69
list(REMOVE_ITEM TEST_OPS test_parallel_executor_crf)
Y
yuyang18 已提交
70
list(REMOVE_ITEM TEST_OPS test_parallel_executor_fetch_feed)
71
list(REMOVE_ITEM TEST_OPS test_dist_se_resnext)
72
list(REMOVE_ITEM TEST_OPS test_dgc_op)
73
list(REMOVE_ITEM TEST_OPS test_dist_se_resnext_nccl)
X
Xin Pan 已提交
74
list(REMOVE_ITEM TEST_OPS test_dist_transformer)
75
list(REMOVE_ITEM TEST_OPS test_parallel_executor_transformer)
D
dengkaipeng 已提交
76 77
list(REMOVE_ITEM TEST_OPS test_bilinear_interp_op)
list(REMOVE_ITEM TEST_OPS test_nearest_interp_op)
78
list(REMOVE_ITEM TEST_OPS test_imperative_resnet)
Y
Yan Xu 已提交
79
list(REMOVE_ITEM TEST_OPS test_imperative_se_resnext)
M
minqiyang 已提交
80
list(REMOVE_ITEM TEST_OPS test_imperative_mnist)
D
dzhwinter 已提交
81
list(REMOVE_ITEM TEST_OPS test_ir_memory_optimize_transformer)
L
lujun 已提交
82
list(REMOVE_ITEM TEST_OPS test_layers)
Y
yuyang18 已提交
83 84 85
foreach(TEST_OP ${TEST_OPS})
    py_test_modules(${TEST_OP} MODULES ${TEST_OP})
endforeach(TEST_OP)
86
py_test_modules(test_adam_op_multi_thread MODULES test_adam_op ENVS FLAGS_inner_op_parallelism=4)
Y
yuyang18 已提交
87
py_test_modules(test_warpctc_op MODULES test_warpctc_op ENVS FLAGS_warpctc_dir=${WARPCTC_LIB_DIR} SERIAL)
D
dengkaipeng 已提交
88 89
py_test_modules(test_bilinear_interp_op MODULES test_bilinear_interp_op SERIAL)
py_test_modules(test_nearest_interp_op MODULES test_nearest_interp_op SERIAL)
90
py_test_modules(test_imperative_resnet MODULES test_imperative_resnet ENVS
G
gongweibao 已提交
91
    FLAGS_cudnn_deterministic=1 SERIAL)
M
minqiyang 已提交
92
py_test_modules(test_imperative_mnist MODULES test_imperative_mnist ENVS
G
gongweibao 已提交
93
    FLAGS_cudnn_deterministic=1 SERIAL)
Y
Yan Xu 已提交
94
py_test_modules(test_imperative_se_resnext MODULES test_imperative_se_resnext ENVS
G
gongweibao 已提交
95
    FLAGS_cudnn_deterministic=1 SERIAL)
Y
Yan Xu 已提交
96

97 98 99
if(WITH_DISTRIBUTE)
    py_test_modules(test_dist_train MODULES test_dist_train SERIAL)
    set_tests_properties(test_listen_and_serv_op PROPERTIES TIMEOUT 20)
G
gongweibao 已提交
100 101 102
    if(WITH_DGC)
        py_test_modules(test_dgc_op MODULES test_dgc_op)
    endif()
J
JiabinYang 已提交
103
    if(NOT APPLE)
104 105
        set_tests_properties(test_dist_mnist PROPERTIES TIMEOUT 200)
        set_tests_properties(test_dist_word2vec PROPERTIES TIMEOUT 200)
106 107
        py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext)
        set_tests_properties(test_dist_se_resnext PROPERTIES TIMEOUT 1000)
G
gongweibao 已提交
108
        py_test_modules(test_dist_se_resnext_nccl MODULES test_dist_se_resnext_nccl SERIAL)
109
        set_tests_properties(test_dist_se_resnext_nccl PROPERTIES TIMEOUT 1000)
110
        # FIXME(typhoonzero): add these tests back
111 112
        # py_test_modules(test_dist_transformer MODULES test_dist_transformer)
        # set_tests_properties(test_dist_transformer PROPERTIES TIMEOUT 1000)
W
Wu Yi 已提交
113
        set_tests_properties(test_dist_ctr test_dist_mnist test_dist_mnist_batch_merge test_dist_save_load test_dist_se_resnext test_dist_simnet_bow test_dist_text_classification test_dist_train test_dist_word2vec PROPERTIES RUN_SERIAL TRUE)
114
    endif(NOT APPLE)
115
    # py_test_modules(test_dist_transpiler MODULES test_dist_transpiler)
116
endif()
117

L
Luo Tao 已提交
118 119
py_test_modules(test_parallel_executor_crf MODULES test_parallel_executor_crf SERIAL)
py_test_modules(test_parallel_executor_fetch_feed MODULES test_parallel_executor_fetch_feed SERIAL)
C
chengduo 已提交
120
set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 450)
121
set_tests_properties(test_parallel_executor_seresnext PROPERTIES TIMEOUT 740)
122
py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer SERIAL)
123
py_test_modules(test_layers MODULES test_layers ENVS FLAGS_cudnn_deterministic=1)
D
dzhwinter 已提交
124
if(NOT WIN32)
125
    py_test_modules(test_ir_memory_optimize_transformer MODULES test_ir_memory_optimize_transformer SERIAL)
D
dzhwinter 已提交
126
endif()
127

128
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
C
chengduo 已提交
129 130
    # 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)
131
endif()
132 133 134 135

if (WITH_NGRAPH)
    add_subdirectory(ngraph)
endif()
136 137 138 139

if (WITH_MKLDNN)
    add_subdirectory(mkldnn)
endif()