You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.5 KiB
71 lines
2.5 KiB
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)
|
|
set(op_common_deps operator op_registry)
|
|
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)
|
|
list(LENGTH op_library_DEPS dep_len)
|
|
if (${cu_srcs_len} EQUAL 0 AND ${dep_len} EQUAL 0)
|
|
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()
|
|
|
|
add_subdirectory(math)
|
|
cc_test(gather_test SRCS gather_test.cc DEPS tensor)
|
|
|
|
cc_test(scatter_test SRCS scatter_test.cc DEPS tensor)
|
|
|
|
cc_library(net_op SRCS net_op.cc DEPS op_registry)
|
|
cc_test(net_op_test SRCS net_op_test.cc DEPS net_op)
|
|
|
|
op_library(add_op SRCS add_op.cc add_op.cu)
|
|
|
|
op_library(mean_op SRCS mean_op.cc mean_op.cu)
|
|
|
|
op_library(mul_op SRCS mul_op.cc mul_op.cu DEPS math_function)
|
|
op_library(rowwise_add_op SRCS rowwise_add_op.cu rowwise_add_op.cc)
|
|
|
|
op_library(sigmoid_op SRCS sigmoid_op.cc sigmoid_op.cu)
|
|
op_library(softmax_op SRCS softmax_op.cc softmax_op.cu)
|
|
op_library(gaussian_random_op SRCS gaussian_random_op.cc gaussian_random_op.cu)
|
|
op_library(cross_entropy_op SRCS cross_entropy_op.cc cross_entropy_op.cu)
|
|
op_library(fill_zeros_like_op SRCS fill_zeros_like_op.cc fill_zeros_like_op.cu)
|
|
|
|
op_library(sgd_op SRCS sgd_op.cc sgd_op.cu)
|
|
|
|
op_library(recurrent_op SRCS recurrent_op.cc rnn/recurrent_op_utils.cc
|
|
DEPS framework_proto tensor op_registry operator net_op)
|
|
op_library(uniform_random_op
|
|
SRCS uniform_random_op.cc uniform_random_op.cu)
|