CMakeLists.txt 2.9 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 18 19 20 21
lite_cc_library(subgraph_bridge_stack_op_xpu SRCS stack_op.cc DEPS ${xpu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_gather_op_xpu SRCS gather_op.cc DEPS ${xpu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_scale_op_xpu SRCS scale_op.cc DEPS ${xpu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_lookup_table_op_xpu SRCS lookup_table_op.cc DEPS ${xpu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_slice_op_xpu SRCS slice_op.cc DEPS ${xpu_subgraph_bridge_deps})
22
lite_cc_library(subgraph_bridge_transpose_op_xpu SRCS transpose_op.cc DEPS ${xpu_subgraph_bridge_deps})
23
lite_cc_library(subgraph_bridge_reshape_op_xpu SRCS reshape_op.cc DEPS ${xpu_subgraph_bridge_deps})
24
lite_cc_library(subgraph_bridge_layer_norm_op_xpu SRCS layer_norm_op.cc DEPS ${xpu_subgraph_bridge_deps})
25
lite_cc_library(subgraph_bridge_dropout_op_xpu SRCS dropout_op.cc DEPS ${xpu_subgraph_bridge_deps})
26
lite_cc_library(subgraph_bridge_matmul_op_xpu SRCS matmul_op.cc DEPS ${xpu_subgraph_bridge_deps})
27

28 29 30 31 32 33 34 35 36 37 38
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
39 40 41 42 43
        subgraph_bridge_stack_op_xpu
        subgraph_bridge_gather_op_xpu
        subgraph_bridge_scale_op_xpu
        subgraph_bridge_lookup_table_op_xpu
        subgraph_bridge_slice_op_xpu
44
        subgraph_bridge_transpose_op_xpu
45
        subgraph_bridge_reshape_op_xpu
46
        subgraph_bridge_layer_norm_op_xpu
47
        subgraph_bridge_dropout_op_xpu
48
        subgraph_bridge_matmul_op_xpu
49
        CACHE INTERNAL "xpu_subgraph_bridges")
50

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