CMakeLists.txt 4.9 KB
Newer Older
Z
zhangjinchao01 已提交
1 2 3 4
# gserver pacakge unittests

################### test_ProtoDataProvider ############
add_unittest_without_exec(test_ProtoDataProvider
5
    test_ProtoDataProvider.cpp)
Z
zhangjinchao01 已提交
6 7 8 9 10 11 12 13 14 15 16

# test_ProtoDataProvider will mkdir as same name,
# so if WORKING_DIRECTORY is default directory, then
# mkdir will get error.
add_test(NAME test_ProtoDataProvider
    COMMAND ${CMAKE_CURRENT_BINARY_DIR}/test_ProtoDataProvider
    WORKING_DIRECTORY ${PROJ_ROOT}/paddle)

################# test_LayerGrad #######################
add_unittest_without_exec(test_LayerGrad
    test_LayerGrad.cpp
17
    LayerGradUtil.cpp)
Z
zhangjinchao01 已提交
18
add_test(NAME test_LayerGrad
19
    COMMAND test_LayerGrad)
Z
zhangjinchao01 已提交
20

21 22 23
################ test_CRFLayerGrad ####################
add_unittest_without_exec(test_CRFLayerGrad
    test_CRFLayerGrad.cpp
P
Peng Li 已提交
24
    LayerGradUtil.cpp)
25 26 27
add_test(NAME test_CRFLayerGrad
    COMMAND test_CRFLayerGrad)

C
caoying03 已提交
28 29 30 31 32 33
################ test_SeqSliceLayerGrad ####################
add_unittest_without_exec(test_SeqSliceLayerGrad
    test_SeqSliceLayerGrad.cpp
    LayerGradUtil.cpp)
add_test(NAME test_SeqSliceLayerGrad
    COMMAND test_SeqSliceLayerGrad)
34

35 36
add_unittest_without_exec(test_ActivationGrad
    test_ActivationGrad.cpp
37
    LayerGradUtil.cpp)
38 39
add_test(NAME test_ActivationGrad
    COMMAND test_ActivationGrad)
40 41 42
################# test_ConvTrans #######################
add_unittest_without_exec(test_ConvTrans
    test_ConvTrans.cpp
43
    LayerGradUtil.cpp)
44 45 46

add_test(NAME test_ConvTrans
    COMMAND test_ConvTrans)
G
gaoyuan 已提交
47 48 49
################# test_PriorBox #######################
add_unittest_without_exec(test_PriorBox
    test_PriorBox.cpp
50
    LayerGradUtil.cpp)
G
gaoyuan 已提交
51 52 53

add_test(NAME test_PriorBox
    COMMAND test_PriorBox)
54 55 56 57 58
################# test_DetectionOutput #######################
add_unittest_without_exec(test_DetectionOutput
    test_DetectionOutput.cpp
    LayerGradUtil.cpp)

X
xuwei06 已提交
59
add_test(NAME test_DetectionOutput
60
    COMMAND test_DetectionOutput)
W
wangyang59 已提交
61 62 63
################# test_ConvUnify #######################
add_unittest_without_exec(test_ConvUnify
    test_ConvUnify.cpp
64
    LayerGradUtil.cpp)
65

W
wangyang59 已提交
66 67
add_test(NAME test_ConvUnify
    COMMAND test_ConvUnify)
68 69 70
################# test_BatchNorm #######################
add_unittest_without_exec(test_BatchNorm
    test_BatchNorm.cpp
71
    LayerGradUtil.cpp)
72 73 74

add_test(NAME test_BatchNorm
    COMMAND test_BatchNorm)
C
caoying03 已提交
75 76 77 78 79 80 81 82 83 84


################# test_KmaxSeqScore #######################
add_unittest_without_exec(test_KmaxSeqScore
    test_KmaxSeqScore.cpp
    LayerGradUtil.cpp)

add_test(NAME test_KmaxSeqScore
    COMMAND test_KmaxSeqScore)

Z
zhangjinchao01 已提交
85 86
################## test_Evaluator #######################
add_unittest(test_Evaluator
87
    test_Evaluator.cpp)
Z
zhangjinchao01 已提交
88 89 90 91 92 93 94 95 96 97

################ test_LinearChainCRF ####################
add_simple_unittest(test_LinearChainCRF)

############## test_MultinomialSampler ###################
add_simple_unittest(test_MultinomialSampler)

############## test_PyDataProvider ########################
if(WITH_PYTHON)
    add_unittest_without_exec(test_PyDataProvider
98
        test_PyDataProvider.cpp)
Z
zhangjinchao01 已提交
99 100 101 102 103 104 105

    add_test(NAME test_PyDataProvider
        COMMAND .set_python_path.sh -d ./gserver/tests:${PROJ_ROOT}/python/ ${CMAKE_CURRENT_BINARY_DIR}/test_PyDataProvider
        WORKING_DIRECTORY ${PROJ_ROOT}/paddle)
endif()

############### test_RecurrentLayer #######################
106
add_simple_unittest(test_RecurrentLayer)
Z
zhangjinchao01 已提交
107

108 109
############### test_WarpCTCLayer #######################
if(NOT WITH_DOUBLE)
110
    add_unittest_without_exec(test_WarpCTCLayer
111
        test_WarpCTCLayer.cpp)
112 113

    add_test(NAME test_WarpCTCLayer
L
liaogang 已提交
114
        COMMAND ${CMAKE_CURRENT_BINARY_DIR}/test_WarpCTCLayer --warpctc_dir=${WARPCTC_LIB_DIR}
115
        WORKING_DIRECTORY ${PROJ_ROOT}/paddle)
116 117
endif()

Z
zhangjinchao01 已提交
118 119 120 121 122 123 124 125 126 127 128 129
############### test_RecurrentGradientMachine ###############
# TODO(yuyang18): There is some bug in test_RecurrentGradientMachine
# I will fix it.
add_unittest_without_exec(test_RecurrentGradientMachine
    test_RecurrentGradientMachine.cpp)
add_test(NAME test_RecurrentGradientMachine
    COMMAND .set_python_path.sh -d
            ${PROJ_ROOT}/python:${PROJ_ROOT}/paddle/gserver/tests
            ${CMAKE_CURRENT_BINARY_DIR}/test_RecurrentGradientMachine
    WORKING_DIRECTORY ${PROJ_ROOT}/paddle)

add_unittest_without_exec(test_NetworkCompare
130
    test_NetworkCompare.cpp)
Z
zhangjinchao01 已提交
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
if(WITH_GPU)
    add_test(NAME test_NetworkCompare
        COMMAND .set_python_path.sh -d ${PROJ_ROOT}/python ${CMAKE_CURRENT_BINARY_DIR}/test_NetworkCompare --use_gpu=true
        WORKING_DIRECTORY ${PROJ_ROOT}/paddle)
else()
    add_test(NAME test_NetworkCompare
        COMMAND .set_python_path.sh -d ${PROJ_ROOT}/python ${CMAKE_CURRENT_BINARY_DIR}/test_NetworkCompare --use_gpu=false
        WORKING_DIRECTORY ${PROJ_ROOT}/paddle)
endif()


add_unittest_without_exec(test_PyDataProvider2
        test_PyDataProvider2.cpp)

add_test(NAME test_PyDataProvider2
   COMMAND .set_python_path.sh -d ${PROJ_ROOT}/paddle/gserver/tests:${PROJ_ROOT}/python ${CMAKE_CURRENT_BINARY_DIR}/test_PyDataProvider2
        WORKING_DIRECTORY ${PROJ_ROOT}/paddle
)