CMakeLists.txt 2.2 KB
Newer Older
Z
zhangjinchao01 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
# Gserver package contains:
#   * Layers
#   * Activations
#   * DataProviders
#   * Evaluators
#   * GradientMachines(NeuralNetwork)
file(GLOB_RECURSE GSERVER_HEADER RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "*.h")
file(GLOB_RECURSE GSERVER_SOURCES RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "*.cpp")
set(GSERVER_SOURCES
    layers/LstmCompute.cu
    layers/GruCompute.cu
    ${GSERVER_SOURCES})

macro(filter_test VAR_NAME)
    set(tmp)
    foreach(p IN LISTS ${VAR_NAME})
        if(NOT ${p} MATCHES ".*tests/.*")
             set(tmp ${p} ${tmp})
        endif()
    endforeach()
    set(${VAR_NAME} ${tmp})
endmacro()

filter_test(GSERVER_HEADER)
filter_test(GSERVER_SOURCES)
T
tensor-tang 已提交
26 27

if(NOT WITH_MKLDNN)
28 29
    file(GLOB_RECURSE DNN_HEADER RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "MKLDNN*.h")
    file(GLOB_RECURSE DNN_SOURCES RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "MKLDNN*.cpp")
T
tensor-tang 已提交
30 31
    list(REMOVE_ITEM GSERVER_HEADER ${DNN_HEADER})
    list(REMOVE_ITEM GSERVER_SOURCES ${DNN_SOURCES})
32
    message(STATUS "Skip compiling with MKLDNNLayers and MKLDNNActivations")
T
tensor-tang 已提交
33
else()
34
    message(STATUS "Compile with MKLDNNLayers and MKLDNNActivations")
T
tensor-tang 已提交
35 36
endif()

Z
zhangjinchao01 已提交
37 38
if(NOT WITH_GPU)
    list(REMOVE_ITEM GSERVER_HEADER
W
wangyang59 已提交
39
        layers/CudnnConvBaseLayer.h
Z
zhangjinchao01 已提交
40
        layers/CudnnConvLayer.h
W
wangyang59 已提交
41
        layers/CudnnConvTransLayer.h
Z
zhangjinchao01 已提交
42
        layers/CudnnPoolLayer.h
H
hedaoyuan 已提交
43
        layers/CudnnBatchNormLayer.h)
Z
zhangjinchao01 已提交
44 45

    list(REMOVE_ITEM GSERVER_SOURCES
W
wangyang59 已提交
46
        layers/CudnnConvBaseLayer.cpp
Z
zhangjinchao01 已提交
47
        layers/CudnnConvLayer.cpp
W
wangyang59 已提交
48
        layers/CudnnConvTransLayer.cpp
Z
zhangjinchao01 已提交
49
        layers/CudnnPoolLayer.cpp
H
hedaoyuan 已提交
50
        layers/CudnnBatchNormLayer.cpp)
Z
zhangjinchao01 已提交
51 52 53 54 55 56 57 58 59 60 61 62 63
    compile_cu_as_cpp(layers/LstmCompute.cu)
    compile_cu_as_cpp(layers/GruCompute.cu)
endif()

if(NOT WITH_PYTHON)
    list(REMOVE_ITEM GSERVER_SOURCES
            dataproviders/PyDataProvider.cpp)
    
    list(REMOVE_ITEM GSERVER_HEADER
            dataproviders/PyDataProvider.h)
endif()

if(WITH_GPU)
H
hedaoyuan 已提交
64
    cuda_add_library(paddle_gserver ${GSERVER_SOURCES})
Z
zhangjinchao01 已提交
65 66 67 68 69 70 71
else()
    add_library(paddle_gserver STATIC
        ${GSERVER_SOURCES})
endif()

add_style_check_target(paddle_gserver ${GSERVER_SOURCES})
add_style_check_target(paddle_gserver ${GSERVER_HEADER})
Y
Yu Yang 已提交
72
add_dependencies(paddle_gserver paddle_proto ${external_project_dependencies})
Z
zhangjinchao01 已提交
73 74 75
if(WITH_TESTING)
    add_subdirectory(tests)
endif()