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

M
Commit  
mozga-intel 已提交
4
# The MKLDNN tests are skiped when the MKLDNN flag is OFF
5
if(NOT WITH_MKLDNN)
M
Commit  
mozga-intel 已提交
6 7 8 9 10
    foreach(src ${TEST_OPS})
        if(${src} MATCHES ".*_mkldnn_op$")
            list(REMOVE_ITEM TEST_OPS ${src})
        endif()
    endforeach()
11 12
endif(NOT WITH_MKLDNN)

13 14
if(NOT WITH_DISTRIBUTE)
    list(REMOVE_ITEM TEST_OPS test_recv_op)
15 16 17 18 19
    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 已提交
20
    LIST(REMOVE_ITEM TEST_OPS test_dist_ctr)
M
minqiyang 已提交
21 22 23
    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)
24 25
endif(NOT WITH_DISTRIBUTE)

Y
Yu Yang 已提交
26 27 28
if (NOT ${WITH_GPU})
    LIST(REMOVE_ITEM TEST_OPS test_conv2d_fusion_op)
elseif(${CUDNN_MAJOR_VERSION} VERSION_LESS 7)
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
list(REMOVE_ITEM TEST_OPS test_lstm_unit_op) # # FIXME(qijun) https://github.com/PaddlePaddle/Paddle/issues/5185
Y
yuyang18 已提交
35
list(REMOVE_ITEM TEST_OPS test_nce) # FIXME(qijun) https://github.com/PaddlePaddle/Paddle/issues/7778
36 37 38 39 40
list(REMOVE_ITEM TEST_OPS test_recurrent_op) # FIXME(qijun) https://github.com/PaddlePaddle/Paddle/issues/6152
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
41
if(APPLE)
42 43 44 45
    if(NOT WITH_DISTRIBUTE)
        list(REMOVE_ITEM TEST_OPS test_desc_clone)
        list(REMOVE_ITEM TEST_OPS test_program_code)
    endif(NOT WITH_DISTRIBUTE)
J
Jiabin Yang 已提交
46
    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")
47 48
    # this op is not support on mac
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
49 50 51
    # 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)
52
    list(REMOVE_ITEM TEST_OPS test_fuse_elewise_add_act_pass)
53
endif()
T
Tao Luo 已提交
54 55 56 57
if(NOT WITH_MKLML)
    # this op is not support on openblas
    list(REMOVE_ITEM TEST_OPS test_fusion_seqexpand_concat_fc_op)
endif()
58 59 60

function(py_test_modules TARGET_NAME)
  if(WITH_TESTING)
Y
yuyang18 已提交
61
    set(options SERIAL)
62
    set(oneValueArgs "")
63
    set(multiValueArgs MODULES DEPS ENVS)
64 65
    cmake_parse_arguments(py_test_modules "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
    add_test(NAME ${TARGET_NAME}
66
             COMMAND env PYTHONPATH=${PADDLE_BINARY_DIR}/python ${py_test_modules_ENVS}
67
             ${PYTHON_EXECUTABLE} ${PADDLE_SOURCE_DIR}/tools/test_runner.py ${py_test_modules_MODULES}
68
             WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR})
Y
yuyang18 已提交
69
    if (py_test_modules_SERIAL)
X
Xin Pan 已提交
70
        set_property(TEST ${TARGET_NAME} PROPERTY RUN_SERIAL 1)
Y
yuyang18 已提交
71
    endif()
X
Xin Pan 已提交
72
    set_tests_properties(${TARGET_NAME} PROPERTIES TIMEOUT 600)
73 74 75
  endif()
endfunction()
list(REMOVE_ITEM TEST_OPS test_warpctc_op)
T
typhoonzero 已提交
76
list(REMOVE_ITEM TEST_OPS test_dist_train)
T
typhoonzero 已提交
77
list(REMOVE_ITEM TEST_OPS test_dist_transpiler)
Y
yuyang18 已提交
78
list(REMOVE_ITEM TEST_OPS test_parallel_executor_crf)
Y
yuyang18 已提交
79
list(REMOVE_ITEM TEST_OPS test_parallel_executor_fetch_feed)
80
list(REMOVE_ITEM TEST_OPS test_dist_se_resnext)
X
Xin Pan 已提交
81
list(REMOVE_ITEM TEST_OPS test_dist_transformer)
82 83
list(REMOVE_ITEM TEST_OPS test_parallel_executor_transformer)
list(REMOVE_ITEM TEST_OPS test_image_classification_resnet)
84
list(REMOVE_ITEM TEST_OPS test_interpolate_op)
Y
yuyang18 已提交
85 86 87
foreach(TEST_OP ${TEST_OPS})
    py_test_modules(${TEST_OP} MODULES ${TEST_OP})
endforeach(TEST_OP)
Y
yuyang18 已提交
88
py_test_modules(test_warpctc_op MODULES test_warpctc_op ENVS FLAGS_warpctc_dir=${WARPCTC_LIB_DIR} SERIAL)
89
py_test_modules(test_interpolate_op MODULES test_interpolate_op SERIAL)
90 91 92
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)
J
JiabinYang 已提交
93
    if(NOT APPLE)
94 95
        set_tests_properties(test_dist_mnist PROPERTIES TIMEOUT 200)
        set_tests_properties(test_dist_word2vec PROPERTIES TIMEOUT 200)
W
Wu Yi 已提交
96 97
        py_test_modules(test_dist_se_resnext MODULES test_dist_se_resnext)
        set_tests_properties(test_dist_se_resnext PROPERTIES TIMEOUT 1000)
Y
yi.wu 已提交
98 99 100
        # FIXME(typhoonzero): add this back
	#py_test_modules(test_dist_transformer MODULES test_dist_transformer)
	#set_tests_properties(test_dist_transformer PROPERTIES TIMEOUT 1000)
101 102
        # TODO(typhoonzero): make dist test parallel when fix port management issue
        set_tests_properties(test_dist_mnist test_dist_word2vec test_dist_se_resnext test_dist_ctr test_dist_simnet_bow test_dist_save_load test_dist_text_classification test_dist_mnist_batch_merge PROPERTIES RUN_SERIAL TRUE)
103
    endif(NOT APPLE)
T
typhoonzero 已提交
104
    py_test_modules(test_dist_transpiler MODULES test_dist_transpiler)
105
endif()
L
Luo Tao 已提交
106 107
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)
M
minqiyang 已提交
108
set_tests_properties(test_parallel_executor_fetch_feed PROPERTIES TIMEOUT 150)
109
py_test_modules(test_parallel_executor_transformer MODULES test_parallel_executor_transformer SERIAL)
M
minqiyang 已提交
110 111 112
if(NOT APPLE)
    py_test_modules(test_image_classification_resnet MODULES test_image_classification_resnet SERIAL)
endif()