CMakeLists.txt 1.6 KB
Newer Older
1 2 3
if(NOT LITE_WITH_XPU)
  return()
endif()
4

5 6
lite_cc_library(subgraph_bridge_utility_xpu SRCS utility.cc DEPS ${xpu_builder_libs} tensor)
lite_cc_library(subgraph_bridge_graph_xpu SRCS graph.cc DEPS subgraph_bridge_utility_xpu)
7

8
set(xpu_subgraph_bridge_deps subgraph_bridge_registry subgraph_bridge_utility_xpu subgraph_bridge_graph_xpu)
9

10 11 12 13 14 15 16
lite_cc_library(subgraph_bridge_act_op_xpu SRCS act_op.cc DEPS ${xpu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_conv_op_xpu SRCS conv_op.cc DEPS ${xpu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_elementwise_ops_xpu SRCS elementwise_ops.cc DEPS ${xpu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_pool_op_xpu SRCS pool_op.cc DEPS ${subgraph_bridge_deps_xpu})
lite_cc_library(subgraph_bridge_softmax_op_xpu SRCS softmax_op.cc DEPS ${subgraph_bridge_deps_xpu})
lite_cc_library(subgraph_bridge_mul_op_xpu SRCS mul_op.cc DEPS ${xpu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_batch_norm_op_xpu SRCS batch_norm_op.cc DEPS ${xpu_subgraph_bridge_deps})
17
lite_cc_library(subgraph_bridge_transpose_op_xpu SRCS transpose_op.cc DEPS ${xpu_subgraph_bridge_deps})
18

19 20 21 22 23 24 25 26 27 28 29
set(xpu_subgraph_bridges
        subgraph_bridge_registry
        subgraph_bridge_utility_xpu
        subgraph_bridge_graph_xpu
        subgraph_bridge_act_op_xpu
        subgraph_bridge_conv_op_xpu
        subgraph_bridge_elementwise_ops_xpu
        subgraph_bridge_pool_op_xpu
        subgraph_bridge_softmax_op_xpu
        subgraph_bridge_mul_op_xpu
        subgraph_bridge_batch_norm_op_xpu
30
        subgraph_bridge_transpose_op_xpu
31
        CACHE INTERNAL "xpu_subgraph_bridges")
32

33
message(STATUS "+++++ xpu_subgraph_bridges: ${xpu_subgraph_bridges}")