CMakeLists.txt 3.6 KB
Newer Older
W
Wu Yi 已提交
1
include(operators)
D
dzhwinter 已提交
2

W
Wu Yi 已提交
3 4 5 6 7
# clean cache and pybind_file content first when rebuild
unset(GLOB_OP_LIB CACHE)
unset(OP_LIBRARY CACHE)
set(pybind_file ${PADDLE_BINARY_DIR}/paddle/fluid/pybind/pybind.h CACHE INTERNAL "pybind.h file")
file(WRITE ${pybind_file} "// Generated by the paddle/fluid/operator/CMakeLists.txt.  DO NOT EDIT!\n\n")
Y
Yu Yang 已提交
8

Q
qijun 已提交
9
add_subdirectory(math)
W
Wu Yi 已提交
10 11 12 13 14 15 16 17 18
add_subdirectory(controlflow)
add_subdirectory(csp)
add_subdirectory(detection)
add_subdirectory(elementwise)
add_subdirectory(fused)
add_subdirectory(metrics)
add_subdirectory(optimizers)
add_subdirectory(reduce_ops)
add_subdirectory(sequence_ops)
Q
QI JUN 已提交
19

T
typhoonzero 已提交
20
if(WITH_DISTRIBUTE)
21
    add_subdirectory(distributed)
W
Wu Yi 已提交
22 23
    add_subdirectory(distributed_ops)
endif()
24

W
wopeizl 已提交
25
add_subdirectory(reader)
T
typhoonzero 已提交
26

W
Wu Yi 已提交
27 28
if (NOT WIN32)
    add_subdirectory(nccl)
S
sneaxiy 已提交
29 30
endif()

31
if (WITH_GPU AND TENSORRT_FOUND)
W
Wu Yi 已提交
32
    add_subdirectory(tensorrt)
33
endif()
34

35 36 37 38 39
SET(OP_HEADER_DEPS xxhash)
if (WITH_GPU)
    SET(OP_HEADER_DEPS ${OP_HEADER_DEPS} cub)
endif()

40 41 42 43 44 45 46
SET(OP_PREFETCH_DEPS "")
if (WITH_DISTRIBUTE)
    SET(OP_PREFETCH_DEPS ${OP_PREFETCH_DEPS} parameter_prefetch)
endif()

register_operators(EXCLUDES warpctc_op conv_fusion_op DEPS ${OP_HEADER_DEPS} ${OP_PREFETCH_DEPS})

武毅 已提交
47

D
Dang Qingqing 已提交
48
# warpctc_op needs cudnn 7 above
W
wopeizl 已提交
49
if (WITH_GPU AND NOT WIN32)
W
Wu Yi 已提交
50 51
    if (${CUDNN_MAJOR_VERSION} VERSION_LESS 7)
        op_library(warpctc_op DEPS dynload_warpctc sequence_padding sequence_scale SRCS warpctc_op.cc warpctc_op.cu.cc)
W
Wu Yi 已提交
52 53
    else()
        op_library(warpctc_op DEPS dynload_warpctc sequence_padding sequence_scale)
W
Wu Yi 已提交
54
    endif()
D
Dang Qingqing 已提交
55 56 57 58 59
    # conv_fusion_op needs cudnn 7 above
    if (NOT ${CUDNN_MAJOR_VERSION} VERSION_LESS 7)
        op_library(conv_fusion_op)
        file(APPEND ${pybind_file} "USE_CUDA_ONLY_OP(conv2d_fusion);\n")
    endif()
W
Wu Yi 已提交
60 61
else()
    op_library(warpctc_op DEPS dynload_warpctc sequence_padding sequence_scale)
W
Wu Yi 已提交
62
endif()
63

64
set(COMMON_OP_DEPS ${OP_HEADER_DEPS})
65

W
wopeizl 已提交
66
set(COMMON_OP_DEPS ${COMMON_OP_DEPS} selected_rows_functor selected_rows lod_tensor maxouting unpooling pooling lod_rank_table context_project sequence_pooling executor)
D
dzhwinter 已提交
67
if (NOT WIN32)
W
wopeizl 已提交
68
    set(COMMON_OP_DEPS ${COMMON_OP_DEPS} dynload_warpctc)
W
Wu Yi 已提交
69
endif()
W
wopeizl 已提交
70 71
set(COMMON_OP_DEPS ${COMMON_OP_DEPS} sequence_padding sequence_scale cos_sim_functor memory jit_kernel concat_and_split cross_entropy softmax vol2col im2col sampler)
set(COMMON_OP_DEPS ${COMMON_OP_DEPS} sequence2batch lstm_compute matrix_bit_code gru_compute activation_functions)
W
Wu Yi 已提交
72
if (WITH_GPU)
N
nhzlx 已提交
73
  set(COMMON_OP_DEPS ${COMMON_OP_DEPS} depthwise_conv prelu)
W
Wu Yi 已提交
74
endif()
75

W
Wu Yi 已提交
76 77 78 79 80 81 82 83
# FIXME(typhoonzero): operator deps may not needed.
# op_library(lod_tensor_to_array_op DEPS lod_rank_table_op)
# op_library(array_to_lod_tensor_op DEPS lod_rank_table_op)
# op_library(unsqueeze_op DEPS reshape_op)
# op_library(squeeze_op DEPS reshape_op)
# op_library(flatten_op DEPS reshape_op)
# op_library(unstack_op DEPS stack_op)
# op_library(tensor_array_to_tensor_op DEPS concat_op)
B
baiyf 已提交
84

S
sneaxiy 已提交
85
set(OPERATOR_DEPS ${OPERATOR_DEPS} ${COMMON_OP_DEPS} python pybind)
W
Wu Yi 已提交
86
set(GLOB_OPERATOR_DEPS ${OPERATOR_DEPS} CACHE INTERNAL "Global Op dependencies")
L
Luo Tao 已提交
87

88
cc_test(gather_test SRCS gather_test.cc DEPS tensor)
Y
Yu Yang 已提交
89
cc_test(scatter_test SRCS scatter_test.cc DEPS tensor math_function)
Q
Qiao Longfei 已提交
90
cc_test(beam_search_decode_op_test SRCS beam_search_decode_op_test.cc DEPS lod_tensor)
Y
Yan Chunwei 已提交
91
cc_test(beam_search_op_test SRCS beam_search_op_test.cc DEPS lod_tensor beam_search_op)
Y
Yi Wang 已提交
92
cc_test(strided_memcpy_test SRCS strided_memcpy_test.cc DEPS tensor memory)
Y
Yu Yang 已提交
93
cc_test(save_load_op_test SRCS save_load_op_test.cc DEPS save_op load_op)
94
cc_test(save_load_combine_op_test SRCS save_load_combine_op_test.cc DEPS save_combine_op load_combine_op)
G
gongweibao 已提交
95
nv_test(dropout_op_test SRCS dropout_op_test.cc DEPS dropout_op tensor)
96

W
Wu Yi 已提交
97
set(GLOB_OP_LIB ${OP_LIBRARY} CACHE INTERNAL "Global OP library")