|
|
@ -245,9 +245,17 @@ op_library(channel_send_op DEPS concurrency)
|
|
|
|
op_library(channel_recv_op DEPS concurrency)
|
|
|
|
op_library(channel_recv_op DEPS concurrency)
|
|
|
|
|
|
|
|
|
|
|
|
list(REMOVE_ITEM GENERAL_OPS ${DEPS_OPS})
|
|
|
|
list(REMOVE_ITEM GENERAL_OPS ${DEPS_OPS})
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# The fully connected layer is deleted when the WITH_MKLDNN flag is OFF
|
|
|
|
|
|
|
|
# Because the fully connected layer has only one MKLDNN's operator
|
|
|
|
|
|
|
|
if(NOT WITH_MKLDNN)
|
|
|
|
|
|
|
|
list(REMOVE_ITEM GENERAL_OPS fc_op)
|
|
|
|
|
|
|
|
endif(NOT WITH_MKLDNN)
|
|
|
|
|
|
|
|
|
|
|
|
foreach(src ${GENERAL_OPS})
|
|
|
|
foreach(src ${GENERAL_OPS})
|
|
|
|
op_library(${src})
|
|
|
|
op_library(${src})
|
|
|
|
endforeach()
|
|
|
|
endforeach()
|
|
|
|
|
|
|
|
|
|
|
|
file(APPEND ${pybind_file} "USE_OP(less_than);\nUSE_OP(logical_and);\nUSE_NO_KERNEL_OP(read_from_array);\n")
|
|
|
|
file(APPEND ${pybind_file} "USE_OP(less_than);\nUSE_OP(logical_and);\nUSE_NO_KERNEL_OP(read_from_array);\n")
|
|
|
|
|
|
|
|
|
|
|
|
add_subdirectory(reader)
|
|
|
|
add_subdirectory(reader)
|
|
|
|