提交 59bbf075 编写于 作者: D dingminghui 提交者: jackzhang235

fix: fix extra bridges not exist in subgraph_mlu

上级 1bf17174
......@@ -102,6 +102,10 @@ if(WITH_GRPC)
add_definitions(-DPADDLE_WITH_GRPC)
endif(WITH_GRPC)
if(LITE_BUILD_EXTRA)
add_definitions(-DLITE_BUILD_EXTRA)
endif()
if(WITH_BRPC_RDMA)
add_definitions(-DPADDLE_WITH_BRPC_RDMA)
endif(WITH_BRPC_RDMA)
......
......@@ -4,9 +4,6 @@ lite_cc_library(subgraph_detector
lite_cc_library(subgraph_pass
SRCS subgraph_pass.cc
DEPS mir_pass types context ${mir_fusers} subgraph_detector)
if (LITE_BUILD_EXTRA)
target_compile_definitions(subgraph_pass PUBLIC "-DLITE_BUILD_EXTRA")
endif()
if (WITH_TESTING AND NOT LITE_WITH_CUDA)
lite_cc_test(test_subgraph_detector
SRCS subgraph_detector_test.cc
......
......@@ -57,11 +57,14 @@ set(mlu_subgraph_bridges
if (LITE_BUILD_EXTRA)
lite_cc_library(subgraph_bridge_lrn_op_mlu SRCS lrn_op.cc DEPS ${subgraph_bridge_deps_mlu})
list(APPEND mlu_subgraph_bridges subgraph_bridge_lrn_op_mlu)
lite_cc_library(subgraph_bridge_gather_op_mlu SRCS gather_op.cc DEPS ${subgraph_bridge_deps_mlu})
list(APPEND mlu_subgraph_bridges subgraph_bridge_gather_op_mlu)
lite_cc_library(subgraph_bridge_norm_op_mlu SRCS norm_op.cc DEPS ${subgraph_bridge_deps_mlu})
list(APPEND mlu_subgraph_bridges subgraph_bridge_norm_op_mlu)
set(mlu_subgraph_bridges
"${mlu_subgraph_bridges}"
subgraph_bridge_lrn_op_mlu
subgraph_bridge_gather_op_mlu
subgraph_bridge_norm_op_mlu
CACHE INTERNAL "mlu_subgraph_bridges")
endif()
lite_cc_library(subgraph_test_helper_mlu SRCS test_helper.cc DEPS ${mlu_subgraph_bridges})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册