CMakeLists.txt 1.5 KB
Newer Older
1 2 3 4 5 6
lite_cc_library(xpu_bridge_registry SRCS registry.cc)

set(xpu_bridge_deps xpu_bridge_registry xpu_builder op)

lite_cc_library(xpu_bridge_act_op SRCS act_op.cc DEPS ${xpu_bridge_deps})
lite_cc_library(xpu_bridge_conv_op SRCS conv_op.cc DEPS ${xpu_bridge_deps})
7 8 9
lite_cc_library(xpu_bridge_elementwise_ops SRCS elementwise_ops.cc DEPS ${xpu_bridge_deps})
lite_cc_library(xpu_bridge_pool_op SRCS pool_op.cc DEPS ${xpu_bridge_deps})
lite_cc_library(xpu_bridge_softmax_op SRCS softmax_op.cc DEPS ${xpu_bridge_deps})
Z
zhupengyang 已提交
10
lite_cc_library(xpu_bridge_mul_op SRCS mul_op.cc DEPS ${xpu_bridge_deps})
11 12 13 14 15

set(xpu_bridges
        xpu_bridge_registry
        xpu_bridge_act_op
        xpu_bridge_conv_op
16 17 18
        xpu_bridge_elementwise_ops
        xpu_bridge_pool_op
        xpu_bridge_softmax_op
Z
zhupengyang 已提交
19
        xpu_bridge_mul_op
20 21 22 23 24 25
        CACHE INTERNAL "xpu_bridges")

set(xpu_bridge_test_deps ${xpu_bridges} ${xpu_kernels} ${ops})

lite_cc_test(test_xpu_bridge_act_op SRCS act_op_test.cc test_helper.cc DEPS ${xpu_bridge_test_deps})
lite_cc_test(test_xpu_bridge_conv_op SRCS conv_op_test.cc test_helper.cc DEPS ${xpu_bridge_test_deps})
26 27 28
lite_cc_test(test_xpu_bridge_elementwise_ops SRCS elementwise_ops_test.cc test_helper.cc DEPS ${xpu_bridge_test_deps})
lite_cc_test(test_xpu_bridge_pool_op SRCS pool_op_test.cc test_helper.cc DEPS ${xpu_bridge_test_deps})
lite_cc_test(test_xpu_bridge_softmax_op SRCS softmax_op_test.cc test_helper.cc DEPS ${xpu_bridge_test_deps})
Z
zhupengyang 已提交
29
lite_cc_test(test_xpu_bridge_mul_op SRCS mul_op_test.cc test_helper.cc DEPS ${xpu_bridge_test_deps})