CMakeLists.txt 5.1 KB
Newer Older
Z
zhangjinchao01 已提交
1 2 3 4
################# test_Compare ############################
add_unittest_without_exec(test_Compare
    test_Compare.cpp)
add_test(NAME test_Compare
5
  COMMAND ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d ${PADDLE_SOURCE_DIR}/python
Z
zhangjinchao01 已提交
6
        ${CMAKE_CURRENT_BINARY_DIR}/test_Compare
7
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)
Z
zhangjinchao01 已提交
8 9 10 11 12

################# test_Trainer ###########################
add_unittest_without_exec(test_Trainer
    test_Trainer.cpp)
add_test(NAME test_Trainer
13 14 15
  COMMAND ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d ${PADDLE_SOURCE_DIR}/python/
        ${PYTHON_EXECUTABLE} ${PADDLE_SOURCE_DIR}/paddle/trainer/tests/gen_proto_data.py &&
        ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d ${PADDLE_SOURCE_DIR}/python/
Z
zhangjinchao01 已提交
16
        ${CMAKE_CURRENT_BINARY_DIR}/test_Trainer
17
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)
Z
zhangjinchao01 已提交
18 19

############### test_TrainerOnePass ##########################
20 21 22 23 24 25
if(WITH_PYTHON)
  # only run test_TrainerOnePass when PYTHON is enabled, because train one pass
  # is using PyDataProvider2.
  add_unittest_without_exec(test_TrainerOnePass
      test_TrainerOnePass.cpp)
  add_test(NAME test_TrainerOnePass
26 27 28 29
    COMMAND  ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d
          ${PADDLE_SOURCE_DIR}/python/:${PADDLE_SOURCE_DIR}/paddle/trainer/tests
          ${PADDLE_SOURCE_DIR}/paddle/.set_port.sh -p port ${CMAKE_CURRENT_BINARY_DIR}/test_TrainerOnePass
      WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)
30
endif()
Z
zhangjinchao01 已提交
31 32 33 34
################ test_CompareTwoNets ######################
add_unittest_without_exec(test_CompareTwoNets
    test_CompareTwoNets.cpp)
add_test(NAME test_CompareTwoNets
35
  COMMAND ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d ${PADDLE_SOURCE_DIR}/python/
Z
zhangjinchao01 已提交
36 37
        ${CMAKE_CURRENT_BINARY_DIR}/test_CompareTwoNets
            --config_file_a=trainer/tests/sample_trainer_config_qb_rnn.conf --config_file_b=trainer/tests/sample_trainer_config_rnn.conf
38
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)
Z
zhangjinchao01 已提交
39

40 41
################ test_CompareMKLDNNandCPU ######################
if(WITH_MKLDNN)
T
tensor-tang 已提交
42 43 44 45 46 47 48 49 50 51 52 53
  macro(gen_command VAR_NAME CONFIG_FILE)
    set(${VAR_NAME} "${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh" "-d" "${PADDLE_SOURCE_DIR}/python/"
                    "${CMAKE_CURRENT_BINARY_DIR}/test_CompareMKLDNNandCPU --use_gpu=False"
                    "--config_file_a=trainer/tests/${CONFIG_FILE} --use_mkldnn_a=True"
                    "--config_file_b=trainer/tests/${CONFIG_FILE} --use_mkldnn_b=False"
                    "WORKING_DIRECTORY" "${PADDLE_SOURCE_DIR}/paddle/")
  endmacro()
  add_unittest_without_exec(test_CompareMKLDNNandCPU test_CompareTwoNets.cpp)
  gen_command(compare_simple_net "sample_trainer_config_simple_net.conf")
  gen_command(compare_branch_net "sample_trainer_config_branch_net.conf")
  add_test(NAME test_CompareMKLDNNandCPU_simple_net COMMAND ${compare_simple_net})
  add_test(NAME test_CompareMKLDNNandCPU_branch_net COMMAND ${compare_branch_net})
54 55
endif()

Z
zhangjinchao01 已提交
56 57 58 59
############### test_CompareTwoOpts ###################
add_unittest_without_exec(test_CompareTwoOpts
    test_CompareTwoOpts.cpp)
add_test(NAME test_CompareTwoOpts
60
  COMMAND ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d ${PADDLE_SOURCE_DIR}/python/
Z
zhangjinchao01 已提交
61 62
        ${CMAKE_CURRENT_BINARY_DIR}/test_CompareTwoOpts
            --config_file_a=trainer/tests/sample_trainer_config_opt_a.conf --config_file_b=trainer/tests/sample_trainer_config_opt_b.conf
63
            --num_passes=1 --need_high_accuracy=0
64
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)
Z
zhangjinchao01 已提交
65 66 67 68

################# test_CompareSparse ##################
add_unittest_without_exec(test_CompareSparse
    test_CompareSparse.cpp)
Y
Yu Yang 已提交
69 70
if(NOT ON_TRAVIS)
  add_test(NAME test_CompareSparse
71
    COMMAND ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d ${PADDLE_SOURCE_DIR}/python/
Y
Yu Yang 已提交
72 73
              ./.set_port.sh -p port -n 6
                  ${CMAKE_CURRENT_BINARY_DIR}/test_CompareSparse
74
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)
Y
Yu Yang 已提交
75
endif()
Z
zhangjinchao01 已提交
76 77 78 79
################# test_recurrent_machine_generation ###############
add_unittest_without_exec(test_recurrent_machine_generation
    test_recurrent_machine_generation.cpp)
add_test(NAME test_recurrent_machine_generation
80
  COMMAND ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d ${PADDLE_SOURCE_DIR}/python/
Z
zhangjinchao01 已提交
81
        ${CMAKE_CURRENT_BINARY_DIR}/test_recurrent_machine_generation
82
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)
Z
zhangjinchao01 已提交
83 84 85 86 87 88

#################### test_PyDataProviderWrapper #########################
add_unittest_without_exec(test_PyDataProviderWrapper
    test_PyDataProviderWrapper.cpp)

add_test(NAME test_PyDataProviderWrapper
89 90
  COMMAND ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d
        ${PADDLE_SOURCE_DIR}/python/:${PADDLE_SOURCE_DIR}/paddle/trainer/tests
Z
zhangjinchao01 已提交
91
        ${CMAKE_CURRENT_BINARY_DIR}/test_PyDataProviderWrapper
92
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)
Z
zhangjinchao01 已提交
93 94 95

#################### test_config_parser #########################
add_test(NAME test_config_parser
96 97 98
  COMMAND ${PADDLE_SOURCE_DIR}/paddle/.set_python_path.sh -d ${PADDLE_SOURCE_DIR}/python/
        ${PYTHON_EXECUTABLE} ${PADDLE_SOURCE_DIR}/paddle/trainer/tests/config_parser_test.py
    WORKING_DIRECTORY ${PADDLE_SOURCE_DIR}/paddle/)