CMakeLists.txt 2.5 KB
Newer Older
Y
Yu Yang 已提交
1 2 3 4 5 6
function(op_library TARGET)
    # op_library is a function to create op library. The interface is same as
    # cc_library. But it handle split GPU/CPU code and link some common library
    # for ops.
    set(cc_srcs)
    set(cu_srcs)
Y
Yu Yang 已提交
7
    set(op_common_deps operator op_registry)
Y
Yu Yang 已提交
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
    set(options "")
    set(oneValueArgs "")
    set(multiValueArgs SRCS DEPS)
    cmake_parse_arguments(op_library "${options}" "${oneValueArgs}"
            "${multiValueArgs}" ${ARGN})

    foreach(src ${op_library_SRCS})
        if (${src} MATCHES ".*\\.cu$")
            list(APPEND cu_srcs ${src})
        elseif(${src} MATCHES ".*\\.cc$")
            list(APPEND cc_srcs ${src})
        else()
            message(FATAL_ERROR "${TARGET} Source file ${src} should only be .cc or .cu")
        endif()
    endforeach()

    list(LENGTH cc_srcs cc_srcs_len)
    if (${cc_srcs_len} EQUAL 0)
        message(FATAL_ERROR "The op library ${TARGET} should contains at least one .cc file")
    endif()

    list(LENGTH cu_srcs cu_srcs_len)
Y
Yu Yang 已提交
30 31
    list(LENGTH op_library_DEPS dep_len)
    if (${cu_srcs_len} EQUAL 0 AND ${dep_len} EQUAL 0)
Y
Yu Yang 已提交
32 33 34 35 36 37 38 39 40 41 42 43
        message(WARNING "The op library ${TARGET} not support GPU!")
    endif()

    if (WITH_GPU)
        nv_library(${TARGET} SRCS ${cc_srcs} ${cu_srcs} DEPS ${op_library_DEPS}
                ${op_common_deps})
    else()
        cc_library(${TARGET} SRCS ${cc_srcs} DEPS ${op_library_DEPS}
                ${op_common_deps})
    endif()
endfunction()

Y
Yan Chunwei 已提交
44 45 46
cc_library(net_op SRCS net_op.cc DEPS op_registry)
cc_test(net_op_test SRCS net_op_test.cc DEPS net_op)

Y
Yu Yang 已提交
47
op_library(add_op SRCS add_op.cc add_op.cu)
Y
Yu Yang 已提交
48
cc_test(add_op_test SRCS add_op_test.cc DEPS add_op)
49

L
liaogang 已提交
50 51 52
op_library(mean_op SRCS mean_op.cc mean_op.cu)
cc_test(mean_op_test SRCS mean_op_test.cc DEPS mean_op)

53 54
op_library(mul_op SRCS mul_op.cc mul_op.cu)
op_library(rowwise_add_op SRCS rowwise_add_op.cu rowwise_add_op.cc)
L
liaogang 已提交
55 56

op_library(sigmoid_op SRCS sigmoid_op.cc sigmoid_op.cu)
57
op_library(softmax_op SRCS softmax_op.cc softmax_op.cu)
Q
Qiao Longfei 已提交
58
op_library(cross_entropy_op SRCS cross_entropy_op.cc cross_entropy_op.cu)
59
op_library(fill_zeros_like_op SRCS fill_zeros_like_op.cc fill_zeros_like_op.cu)
Y
Yu Yang 已提交
60

Q
Qiao Longfei 已提交
61
op_library(sgd_op SRCS sgd_op.cc sgd_op.cu)
Y
Yan Chunwei 已提交
62

L
liaogang 已提交
63 64
op_library(fc_op
    SRCS fc_op.cc
Y
Yan Chunwei 已提交
65
    DEPS mul_op rowwise_add_op sigmoid_op softmax_op net_op)
66 67
op_library(recurrent_op SRCS recurrent_op.cc rnn/recurrent_op_utils.cc
    DEPS op_desc tensor op_registry operator net_op)
68
cc_test(recurrent_op_test SRCS recurrent_op_test.cc DEPS recurrent_op gtest mul_op add_op)